Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/ops/
Dreduce_sum.cc51 auto axis_value = GetValue<int64_t>(*it); in InferImplReduceFuncCalShape() local
57 auto axis_value = GetValue<int64_t>(*it); in InferImplReduceFuncCalShape() local
70 int64_t axis_value = GetValue<int64_t>(axis); in InferImplReduceFuncCalShape() local
130 ValuePtr axis_value; in InferShape() local
Dreduce.cc36 …::vector<int64_t> infer_shape_reduce(std::vector<int64_t> input_x_shape, const ValuePtr axis_value, in infer_shape_reduce()
70 auto axis_value = input_args[1]->BuildValue(); in InferShape() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_gather_parser.cc39 auto axis_value = MakeValue(static_cast<int32_t>(tflite_attr->axis)); in Parse() local
/third_party/boost/boost/gil/
Dpoint.hpp225 T const& axis_value(point<T> const& p) in axis_value() function
234 T& axis_value(point<T>& p) in axis_value() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dreduce_min_fission.cc123 auto axis_value = prim->GetAttr(kAttrAxis); in Process() local
/third_party/harfbuzz/src/
Dhb-ot-stat-table.hh303 const AxisValue& axis_value = this+axis_values[i]; in get_value() local
328 const AxisValue &axis_value = (this + get_axis_value_offsets ()[axis_value_index]); in get_axis_value_name_id() local
/third_party/skia/third_party/externals/harfbuzz/src/
Dhb-ot-stat-table.hh303 const AxisValue& axis_value = this+axis_values[i]; in get_value() local
328 const AxisValue &axis_value = (this + get_axis_value_offsets ()[axis_value_index]); in get_axis_value_name_id() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_maths.cc93 int64_t axis_value; in InferImplReduceFuncCalShape() local
111 auto axis_value = GetValue<int64_t>(axis); in InferImplReduceFuncCalShape() local
Dparam_validator.cc152 int64_t axis_value = GetValue<int64_t>(axis); in CheckAxis() local
Dprim_arrays.cc110 int64_t axis_value = CheckAxis(op_name, axis, -(rank_base + 1), rank_base); in InferImplStack() local
844 int64_t axis_value = CheckAxis(op_name, axis, -(rank + 1), rank); in InferImplSplit() local
960 int64_t axis_value = CheckAxis(op_name, axis, -(rank_base + 1), rank_base); in InferImplConcat() local
1082 auto axis_value = GetValue<int64_t>(axis); in InferImplArgMaxWithValue() local
Dutils.cc218 int64_t GetPositiveAxis(int64_t axis_value, size_t increment) { in GetPositiveAxis()
/third_party/flutter/skia/third_party/externals/harfbuzz/src/
Dhb-ot-stat-table.hh278 const AxisValue &axis_value = (this + get_axis_value_offsets ()[axis_value_index]); in get_axis_value_name_id() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_graph_kernel.cc162 auto axis_value = new_prim->GetAttr(kAttrAxis); in UpdateFracNZReduceOp() local
/third_party/weston/tests/
Dweston-test-client-helper.h99 double axis_value; member
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc166 auto axis_value = GetAxis(softmax_output_node); in CreateReduceMean() local
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc378 auto axis_value = axis->BuildValue(); in InferImplReduceShape() local
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc795 auto axis_value = dyn_cast<ValueNode>(input_axis)->value(); in ExportPrimReduce() local