Searched defs:c_axis (Results 1 – 10 of 10) sorted by relevance
69 const size_t c_axis = 1; in ConvolutionGradAscendCustomize() local
63 size_t num, uint64_t n_axis, uint64_t c_axis) { in CheckAttrTupleAndNCDimensions()134 constexpr uint64_t c_axis = 1; in DeformableOffsetsInferShape() local
78 auto c_axis = kInputIndex1; in BNTrainingReduceInferShape() local
92 auto c_axis = format == Format::NHWC ? input_x_shape.size() - kInputIndex1 : kInputIndex1; in BNTrainingUpdateCheckShapes() local
91 uint64_t c_axis = shapeIndex3; in Dilation2DInferShape() local
97 constexpr uint64_t c_axis = 1; in DeformableConv2dInferShape() local
228 uint64_t c_axis = 1; in Conv2dInferShape() local
261 uint64_t c_axis = 1; in InferShape() local
113 auto c_axis = shapeIndex3; in Dilation2DBackpropFilterInferShape() local
114 auto c_axis = shapeIndex3; in Dilation2DBackpropInputInferShape() local