Searched refs:kShape2dDims (Results 1 – 8 of 8) sorted by relevance
35 size_t axis = kShape2dDims - input_shape.size(); in InitKernel()36 if (input_shape.empty() || input_shape.size() > kShape2dDims) { in InitKernel()37 …MS_LOG(EXCEPTION) << "Input shape should not empty or greater than " << kShape2dDims << "-D, but g… in InitKernel()
55 if (bn_shape_i0.size() < kShape2dDims) { in CreateOutputsOfBNTrainingReduce()56 MS_LOG(INFO) << "The BatchNorm's first input's shape dims less than " << kShape2dDims; in CreateOutputsOfBNTrainingReduce()
217 if (shape.size() < kShape2dDims) { in IsBroadCastSupportFracNZ()229 … [](const std::vector<size_t> &elem) { return elem.size() < kShape2dDims; }); in IsBroadCastSupportFracNZ()
256 if (format == kOpFormat_FRAC_NZ && shape.size() > kShape2dDims) { in IsShapeMatchFormat()
56 if (format == kOpFormat_FRAC_NZ && input_shape.size() <= kShape2dDims) { in GetKernelFormat()
645 if (shape.size() < kShape2dDims) { in FracNZDeviceShape()666 if (shape.size() < kShape2dDims) { in FracNZDeviceDynamicShape()718 if (shape.size() < kShape2dDims) {752 if (shape.size() < kShape2dDims) {
494 const size_t kShape2dDims = 2; variable
237 if (weight_infer_shape.size() < kShape2dDims && *default_format == kOpFormat_FRAC_NZ) { in GetDefaultFormat()