Home
last modified time | relevance | path

Searched refs:x_channel (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dbias_add.cc63 …auto x_channel = data_format == Format::NHWC ? input_shape[input_shape.size() - 1] : input_shape[1… in InferShape() local
66 if (x_not_dyn && bias_shape[0] != x_channel) { in InferShape()
68 … << ", got bias_shape[0]: " << bias_shape[0] << ", x_channel: " << x_channel << "."; in InferShape()
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc364 auto x_channel = data_format == Format::NHWC ? x_shape[x_shape.size() - 1] : x_shape[1]; in InferImplBiasAdd() local
368 if (x_not_dyn && bias_shape[0] != x_channel) { in InferImplBiasAdd()
370 … << ", got bias_shape[0]: " << bias_shape[0] << ", x_channel: " << x_channel << "."; in InferImplBiasAdd()