Commit 7d0e91be authored by Louis Feng's avatar Louis Feng

fixed merge errors.

parent 91c9a1bf
...@@ -134,12 +134,12 @@ size_t MKLDNNEmitter::build_convolution_forward(const mkldnn::memory::desc& inpu ...@@ -134,12 +134,12 @@ size_t MKLDNNEmitter::build_convolution_forward(const mkldnn::memory::desc& inpu
mkldnn::memory::dims(padding_above.begin(), padding_above.end()), mkldnn::memory::dims(padding_above.begin(), padding_above.end()),
mkldnn::padding_kind::zero}, mkldnn::padding_kind::zero},
mkldnn_utils::global_cpu_engine}, mkldnn_utils::global_cpu_engine},
*mkldnn_primitives[input_data_index], *m_mkldnn_primitives[input_data_index],
*mkldnn_primitives[weights_index], *m_mkldnn_primitives[weights_index],
*mkldnn_primitives[bias_index], *m_mkldnn_primitives[bias_index],
*mkldnn_primitives[result_index])); *m_mkldnn_primitives[result_index]));
primitive_deps[conv_index] = {input_data_index, weights_index, bias_index, result_index}; m_primitive_deps[conv_index] = {input_data_index, weights_index, bias_index, result_index};
return conv_index; return conv_index;
} }
...@@ -186,11 +186,11 @@ size_t MKLDNNEmitter::build_convolution_backward_data(const mkldnn::memory::desc ...@@ -186,11 +186,11 @@ size_t MKLDNNEmitter::build_convolution_backward_data(const mkldnn::memory::desc
fwd_pd}; fwd_pd};
const size_t conv_index = insert_primitive(new mkldnn::convolution_backward_data(bwd_pd, const size_t conv_index = insert_primitive(new mkldnn::convolution_backward_data(bwd_pd,
*mkldnn_primitives[in_delta_index], *m_mkldnn_primitives[in_delta_index],
*mkldnn_primitives[in_weights_index], *m_mkldnn_primitives[in_weights_index],
*mkldnn_primitives[out_data_delta_index])); *m_mkldnn_primitives[out_data_delta_index]));
primitive_deps[conv_index] = {in_weights_index, in_delta_index, out_data_delta_index}; m_primitive_deps[conv_index] = {in_weights_index, in_delta_index, out_data_delta_index};
return conv_index; return conv_index;
} }
...@@ -239,12 +239,12 @@ size_t MKLDNNEmitter::build_convolution_backward_filters_bias(const mkldnn::memo ...@@ -239,12 +239,12 @@ size_t MKLDNNEmitter::build_convolution_backward_filters_bias(const mkldnn::memo
fwd_pd}; fwd_pd};
const size_t conv_index = insert_primitive(new mkldnn::convolution_backward_weights(bwd_pd, const size_t conv_index = insert_primitive(new mkldnn::convolution_backward_weights(bwd_pd,
*mkldnn_primitives[in_data_index], *m_mkldnn_primitives[in_data_index],
*mkldnn_primitives[in_delta_index], *m_mkldnn_primitives[in_delta_index],
*mkldnn_primitives[out_weights_delta_index], *m_mkldnn_primitives[out_weights_delta_index],
*mkldnn_primitives[out_bias_delta_index])); *m_mkldnn_primitives[out_bias_delta_index]));
primitive_deps[conv_index] = {in_data_index, in_delta_index, out_weights_delta_index, out_bias_delta_index}; m_primitive_deps[conv_index] = {in_data_index, in_delta_index, out_weights_delta_index, out_bias_delta_index};
return conv_index; return conv_index;
} }
......
...@@ -58,8 +58,7 @@ op::ConvolutionBias::ConvolutionBias(const std::shared_ptr<Node>& data_batch, ...@@ -58,8 +58,7 @@ op::ConvolutionBias::ConvolutionBias(const std::shared_ptr<Node>& data_batch,
{ {
} }
std::shared_ptr<Node> op::ConvolutionBias::copy_with_new_args( std::shared_ptr<Node> op::ConvolutionBias::copy_with_new_args(const NodeVector& new_args) const
const std::vector<std::shared_ptr<Node>>& new_args) const
{ {
if (new_args.size() != 2) if (new_args.size() != 2)
{ {
...@@ -199,7 +198,7 @@ op::ConvolutionBiasBackpropData::ConvolutionBiasBackpropData(const Shape& data_b ...@@ -199,7 +198,7 @@ op::ConvolutionBiasBackpropData::ConvolutionBiasBackpropData(const Shape& data_b
} }
std::shared_ptr<Node> op::ConvolutionBiasBackpropData::copy_with_new_args( std::shared_ptr<Node> op::ConvolutionBiasBackpropData::copy_with_new_args(
const std::vector<std::shared_ptr<Node>>& new_args) const const NodeVector& new_args) const
{ {
if (new_args.size() != 2) if (new_args.size() != 2)
{ {
...@@ -299,7 +298,7 @@ op::ConvolutionBiasBackpropFiltersBias::ConvolutionBiasBackpropFiltersBias( ...@@ -299,7 +298,7 @@ op::ConvolutionBiasBackpropFiltersBias::ConvolutionBiasBackpropFiltersBias(
} }
std::shared_ptr<Node> op::ConvolutionBiasBackpropFiltersBias::copy_with_new_args( std::shared_ptr<Node> op::ConvolutionBiasBackpropFiltersBias::copy_with_new_args(
const std::vector<std::shared_ptr<Node>>& new_args) const const NodeVector& new_args) const
{ {
if (new_args.size() != 2) if (new_args.size() != 2)
{ {
......
...@@ -37,8 +37,7 @@ namespace ngraph ...@@ -37,8 +37,7 @@ namespace ngraph
std::shared_ptr<Node> get_bias() { return get_input_op(2); } std::shared_ptr<Node> get_bias() { return get_input_op(2); }
std::shared_ptr<Node> get_filters() { return get_input_op(1); } std::shared_ptr<Node> get_filters() { return get_input_op(1); }
std::shared_ptr<Node> get_data_batch() { return get_input_op(0); } std::shared_ptr<Node> get_data_batch() { return get_input_op(0); }
virtual std::shared_ptr<Node> copy_with_new_args( virtual std::shared_ptr<Node> copy_with_new_args(const NodeVector& new_args) const override;
const std::vector<std::shared_ptr<Node>>& new_args) const override;
void generate_adjoints(autodiff::Adjoints& adjoints, void generate_adjoints(autodiff::Adjoints& adjoints,
const std::shared_ptr<Node>& delta) override; const std::shared_ptr<Node>& delta) override;
...@@ -84,8 +83,7 @@ namespace ngraph ...@@ -84,8 +83,7 @@ namespace ngraph
const CoordinateDiff& padding_above_forward, const CoordinateDiff& padding_above_forward,
const Strides& data_dilation_strides_forward); const Strides& data_dilation_strides_forward);
virtual std::shared_ptr<Node> copy_with_new_args( virtual std::shared_ptr<Node> copy_with_new_args(const NodeVector& new_args) const override;
const std::vector<std::shared_ptr<Node>>& new_args) const override;
/// \return The data batch shape. /// \return The data batch shape.
const Shape& get_data_batch_shape() const { return m_data_batch_shape; } const Shape& get_data_batch_shape() const { return m_data_batch_shape; }
...@@ -180,8 +178,7 @@ namespace ngraph ...@@ -180,8 +178,7 @@ namespace ngraph
const CoordinateDiff& padding_above_forward, const CoordinateDiff& padding_above_forward,
const Strides& data_dilation_strides_forward); const Strides& data_dilation_strides_forward);
virtual std::shared_ptr<Node> copy_with_new_args( virtual std::shared_ptr<Node> copy_with_new_args(const NodeVector& new_args) const override;
const std::vector<std::shared_ptr<Node>>& new_args) const override;
/// \return The filters tensor shape. /// \return The filters tensor shape.
const Shape& get_filters_shape() const { return m_filters_shape; } const Shape& get_filters_shape() const { return m_filters_shape; }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment