Home
last modified time | relevance | path

Searched defs:shape_value (Results 1 – 25 of 35) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dzeros.cc27 auto shape_value = GetArrayValue<int64_t>(input_args[kInputIndex0]); in InferValue() local
Dones.cc27 auto shape_value = GetArrayValue<int64_t>(input_args[kInputIndex0]); in InferValue() local
/third_party/mindspore/mindspore-src/source/tests/ut/python/nn/
Dtest_nn_embedding.py47 def compile_multi_field_embedding(shape_id, shape_value, shape_field, argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Ddropout_gen_mask.cc128 auto shape_value = shape_args->GetValue(); in DropoutGenMaskInferShape() local
150 auto shape_value = shape_args->GetValue(); in DropoutGenMaskInferShape() local
Dcsr_gather.cc59 auto shape_value = sparse_shape->GetValue()->cast<ValueTuplePtr>(); in CSRGatherInfer() local
Dmake_csrtensor.cc74 auto shape_value = shape->GetValue()->cast<ValueTuplePtr>(); in MakeCSRTensorInfer() local
Dfse_decode.cc68 auto shape_value = y_shape->shape(); in InferShape() local
Dtile_size.cc84 auto shape_value = input_args[0]->GetValue(); in InferValue() local
Dlog_space.cc58 int64_t shape_value = GetValue<int64_t>(primitive->GetAttr("steps")); in LogSpaceInferShape() local
Drandom_poisson.cc65 auto shape_value = input_args[kInputIndex0]->GetValue(); in RandomPoissonInferShape() local
Drandom_standard_laplace.cc71 auto shape_value = input_args[kInputIndex0]->GetValue(); in StandardLaplaceInferShape() local
Dparameterized_truncated_normal.cc89 auto shape_value = input_args[kInputIndex0]->GetValue(); in ParameterizedTruncatedNormalInferShape() local
Drandom_standard_normal.cc72 auto shape_value = input_args[kInputIndex0]->GetValue(); in RandomStandardNormalInferShape() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_scatter_nd.cc36 ShapeVector shape_value; member
Dtest_ops_ones.cc29 ValuePtr shape_value; member
Dtest_ops_zeros.cc29 ValuePtr shape_value; member
Dtest_ops_fill.cc30 ValuePtr shape_value; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Drandom_distribute_info.cc160 ValueTuplePtr shape_value = input_value_[0]->cast<ValueTuplePtr>(); in ResetInputsShape() local
Dgamma_info.cc153 ValueTuplePtr shape_value = input_value_[0]->cast<ValueTuplePtr>(); in ResetInputsShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dbroadcast_for_select.cc76 auto shape_value = MakeValue(broad_shape); in AddBroadCastToNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dgraph_builder.cc33 auto shape_value = Tensor(shape); in BroadcastTo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dgamma_cpu_kernel.cc49 const auto *shape_value = reinterpret_cast<T *>(inputs[0]->device_ptr()); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dformat_utils.cc372 auto shape_value = abstract->BuildValue(); in ConvertAbstractFormatShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Ddropout_for_ge.cc91 auto shape_value = CreateShapeValueNode(func_graph, input_shape->shape(), true); in CreateDropoutGenMaskCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc637 std::vector<int64_t> shape_value = casted_shape->shape(); in ExtractNewShapeFromShape() local
688 std::vector<int64_t> shape_value = {0}; in GetNodeNewShape() local
694 std::vector<int64_t> shape_value = {SizeToLong(tuple_shape_ptr->size())}; in GetNodeNewShape() local
703 std::vector<int64_t> shape_value = {-1}; in GetNodeNewShape() local
708 std::vector<int64_t> shape_value = shape_ptr->shape(); in GetNodeNewShape() local
711 std::vector<int64_t> shape_value = {}; in GetNodeNewShape() local
1971 auto shape_value = parallel::ToFullShape(shape_ptr->shape(), index); in ExtendInputArgsAbstractShape() local
1982 auto shape_value = input_shape; in ToFullShape() local
2001 ShapeVector shape_value; in ToFullShape() local

12