Searched refs:dim_ (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | reduce_sum.cc | 27 int64_t dim_ = static_cast<int64_t>(dim); in InferImplReduceFuncCheckAxis() local
28 if (axis < -dim_ || axis >= dim_) { in InferImplReduceFuncCheckAxis()
29 …MS_LOG(EXCEPTION) << "axis should be in [" << -dim_ << ", " << dim_ << "). But got axis = " << axi… in InferImplReduceFuncCheckAxis()
32 if (axis >= -dim_ && axis < 0) { in InferImplReduceFuncCheckAxis()
33 ret_axis += dim_; in InferImplReduceFuncCheckAxis()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | gatherd_info.cc | 54 dim_ = LongToSize(dim); in GetAttrs()
55 MS_LOG(INFO) << name_ << ": The dim is " << dim_; in GetAttrs()
74 if (input_strategy[dim_] != 1 || index_strategy[dim_] != 1) { in CheckStrategy()
181 if (dim_ == 0) { in ReComputeBatchSplitFlagList()
196 input0_split[dim_] = 0; in GenerateOpStrategies()
|
| D | gatherd_info.h | 55 size_t dim_ = 0;
|
| /third_party/skia/third_party/externals/tint/src/sem/ |
| D | texture_type.h | 35 ast::TextureDimension dim() const { return dim_; } in dim()
38 ast::TextureDimension const dim_;
|
| D | texture_type.cc | 22 Texture::Texture(ast::TextureDimension dim) : dim_(dim) {} in Texture()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_maths.cc | 69 int64_t dim_ = static_cast<int64_t>(dim); in InferImplReduceFuncCheckAxis() local
70 if (axis < -dim_ || axis >= dim_) { in InferImplReduceFuncCheckAxis()
71 …MS_LOG(EXCEPTION) << "axis should be in [" << -dim_ << ", " << dim_ << "). But got axis = " << axi… in InferImplReduceFuncCheckAxis()
74 if (axis >= -dim_ && axis < 0) { in InferImplReduceFuncCheckAxis()
75 ret_axis += dim_; in InferImplReduceFuncCheckAxis()
|
| D | prim_arrays.cc | 1070 auto dim_ = static_cast<int64_t>(dim); in InferImplArgMaxWithValue() local
1071 if (axis < -dim_ || axis >= dim_) { in InferImplArgMaxWithValue()
1072 …MS_LOG(EXCEPTION) << "axis should be in [" << -dim_ << ", " << dim_ << "). But got axis = " << axi… in InferImplArgMaxWithValue()
1074 if (axis >= -dim_ && axis < 0) { in InferImplArgMaxWithValue()
1075 axis += dim_; in InferImplArgMaxWithValue()
|
| /third_party/flutter/skia/third_party/externals/spirv-tools/source/opt/ |
| D | types.cpp | 143 dim_(dim), in Image()
156 return dim_ == it->dim_ && depth_ == it->depth_ && arrayed_ == it->arrayed_ && in IsSame()
164 oss << "image(" << sampled_type_->str() << ", " << dim_ << ", " << depth_ in str()
|
| D | types.h | 202 SpvDim dim_; variable
|
| /third_party/skia/third_party/externals/spirv-tools/source/opt/ |
| D | types.cpp | 337 dim_(dimen), in Image()
350 return dim_ == it->dim_ && depth_ == it->depth_ && arrayed_ == it->arrayed_ && in IsSameImpl()
359 oss << "image(" << sampled_type_->str() << ", " << dim_ << ", " << depth_ in str()
368 words->push_back(dim_); in GetExtraHashWords()
|
| D | types.h | 322 SpvDim dim() const { return dim_; } in dim()
337 SpvDim dim_; variable
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | types.cpp | 337 dim_(dimen), in Image()
350 return dim_ == it->dim_ && depth_ == it->depth_ && arrayed_ == it->arrayed_ && in IsSameImpl()
359 oss << "image(" << sampled_type_->str() << ", " << dim_ << ", " << depth_ in str()
368 words->push_back(dim_); in GetExtraHashWords()
|
| D | types.h | 322 SpvDim dim() const { return dim_; } in dim()
337 SpvDim dim_; variable
|
| /third_party/spirv-tools/source/opt/ |
| D | types.cpp | 337 dim_(dimen), in Image()
350 return dim_ == it->dim_ && depth_ == it->depth_ && arrayed_ == it->arrayed_ && in IsSameImpl()
359 oss << "image(" << sampled_type_->str() << ", " << dim_ << ", " << depth_ in str()
368 words->push_back(dim_); in GetExtraHashWords()
|
| D | types.h | 322 SpvDim dim() const { return dim_; } in dim()
337 SpvDim dim_; variable
|