Home
last modified time | relevance | path

Searched defs:axis_value (Results 1 – 25 of 77) sorted by relevance

1234

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dsplit_tensor.cc41 auto axis_value = GetScalarValue<int64_t>(input_args[kIndex2]->GetValue()); in InferShape() local
76 auto axis_value = GetScalarValue<int64_t>(input_args[kIndex2]->GetValue()); in InferType() local
108 auto axis_value = GetScalarValue<int64_t>(input_args[kIndex2]->GetValue()); in CheckValidation() local
Ddct.cc42 auto axis_value = axis_opt.value(); in InferShape() local
105 auto axis_value = axis_opt.value(); in CheckValidation() local
Dsoftmax.cc50 auto axis_value = axis_value_opt.value(); in CheckValidation() local
79 auto axis_value = axis_value_opt.value(); in InferShape() local
Done_hot_ext.cc55 int64_t axis_value = axis_opt.value(); in InferShape() local
115 int64_t axis_value = axis_opt.value(); in InferShape() local
Dargmax_with_value.cc68 auto axis_value = input_args[kInputIndex1]->GetValue(); in InferShape() local
128 const auto &axis_value = input_values[kInputIndex1]; in InferShape() local
Dexpand_dims.cc34 auto axis_value = input_args[kIndex1]->GetValue(); in InferShape() local
Dcummax.cc36 auto axis_value = axis_opt.value(); in InferShape() local
Dcummin.cc36 auto axis_value = axis_opt.value(); in InferShape() local
Dargmax.cc35 auto axis_value = input_args[kInputIndex1]->GetValue(); in InferShape() local
Dlog_softmax.cc50 auto axis_value = axis_opt.value(); in CheckValidation() local
Dcum_prod.cc42 auto axis_value = axis_opt.value(); in InferShape() local
Dcumsum_ext.cc39 auto axis_value = axis_opt.value(); in InferShape() local
Dlog_softmax_grad.cc54 auto axis_value = axis_opt.value(); in CheckValidation() local
Dcum_sum.cc42 auto axis_value = axis_opt.value(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/converter/parser/onnx/
Donnx_layer_norm_parser_test.cc60 auto axis_value = GetValue<int>(primitive_c->GetAttr("axis")); in TestOnnxLayerNormNode1() local
104 auto axis_value = GetValue<int>(primitive_c->GetAttr("axis")); in TestOnnxLayerNormNode2() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dunstack_ext.cc27 const ValuePtr &axis_value, const TypePtr &element_type) { in InferOutputForRankKnown()
96 auto axis_value = input_args[kInputIndex1]->GetValue(); in InferAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/
Dtflite_gather_parser.cc38 auto axis_value = MakeValue(static_cast<int32_t>(tflite_attr->axis)); in Parse() local
/third_party/skia/m133/third_party/externals/harfbuzz/src/
Dhb-ot-stat-table.hh60 static bool axis_value_is_outside_axis_range (hb_tag_t axis_tag, float axis_value, in axis_value_is_outside_axis_range()
113 float axis_value = get_value (); in keep_axis_value() local
168 float axis_value = get_value (); in keep_axis_value() local
227 float axis_value = get_value (); in keep_axis_value() local
300 float axis_value = rec.get_value (); in keep_axis_value() local
489 const AxisValue& axis_value = this+offsetToAxisValueOffsets+axis_values[i]; in get_value() local
514 const AxisValue &axis_value = (this + get_axis_value_offsets ()[axis_value_index]); in get_axis_value_name_id() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dclip_by_norm.cc52 auto axis_value = primitive->GetAttr(kAxis); in CheckAxisValid() local
128 auto axis_value = GetAttr(kAxis); in GetAxis() local
Dsparse_gather_v2.cc63 auto axis_value = axis_value_opt.value().ToVector(); in GetAxis() local
71 auto axis_value = input->GetValue(); in GetAxis() local
Dcsr_reducesum.cc67 int64_t axis_value = GetValue<int64_t>(axis->GetValue()); in CSRReduceSumInfer() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/
Dsqueeze_axis.cc31 auto axis_value = prim->GetAttr(kAttrAxis); in SqueezeAxis() local
Dsplit_fusion.cc45 auto axis_value = axis_node->cast<ValueNodePtr>()->value(); in SplitFusionProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dsplit_mapper.cc44 ValuePtr axis_value = src_prim->GetAttr("axis"); in Mapper() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/
Dprim_maths.cc50 int64_t axis_value = GetValue<int64_t>(*it); in InferImplReduceFuncCalShape() local
69 auto axis_value = GetValue<int64_t>(axis); in InferImplReduceFuncCalShape() local

1234