Searched refs:shape_value (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | scatter_nd.cc | 27 auto shape_value = input_args[kInputIndex2]->BuildValue(); in InferShape() local
28 auto shape_value_element = GetValue<std::vector<int64_t>>(shape_value); in InferShape()
|
| D | zeros.cc | 33 auto shape_value = input_args[0]->BuildValue(); in ZerosInferShape() local
34 …int64_t> out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("shape", shape_value, prim_name); in ZerosInferShape()
|
| D | ones.cc | 32 auto shape_value = input_args[0]->BuildValue(); in OnesInferShape() local
33 …int64_t> out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("shape", shape_value, prim_name); in OnesInferShape()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | uniform_real_info.cc | 42 ValueTuplePtr shape_value = input_value_[0]->cast<ValueTuplePtr>(); in GetAttrs() local
43 MS_EXCEPTION_IF_NULL(shape_value); in GetAttrs()
44 inputs_shape_.push_back(GetValue<Shape>(shape_value)); in GetAttrs()
|
| D | operator_info.cc | 119 int64_t shape_value = sub_input_shape.at(j); in CheckStrategyValue() local
120 if ((shape_value % strategy_value) != 0) { in CheckStrategyValue()
122 …_LOG(DEBUG) << name_ << ": The strategy is " << StrategyToString(stra) << ", shape " << shape_value in CheckStrategyValue()
125 …_LOG(ERROR) << name_ << ": The strategy is " << StrategyToString(stra) << ", shape " << shape_value in CheckStrategyValue()
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_nn_embedding.py | 47 def compile_multi_field_embedding(shape_id, shape_value, shape_field, argument
51 input_value = Tensor(np.ones(shape_value), type_value)
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 508 ValuePtr shape_value = nullptr; in CreateShapeValueNode() local
524 shape_value = shape_tensor; in CreateShapeValueNode()
534 shape_value = std::make_shared<ValueTuple>(dim_values); in CreateShapeValueNode()
537 MS_EXCEPTION_IF_NULL(shape_value); in CreateShapeValueNode()
539 auto shape_value_node = kernel_graph->NewValueNode(abstract, shape_value); in CreateShapeValueNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | dropout_unify_mindir.cc | 156 auto shape_value = CreateShapeValueNode(func_graph, input_shape->shape(), true); in CreateDropoutGenMaskCNode() local
157 dropout_gen_mask_inputs.push_back(shape_value); in CreateDropoutGenMaskCNode()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 1161 shape_value = shape['value']
1162 validator.check_value_type("shape", shape_value, [tuple], self.name)
1163 for i, value in enumerate(shape_value):
1165 out = {'shape': shape_value,
|
| D | nn_ops.py | 1675 for shape_value in out_shape:
1676 if shape_value <= 0:
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 2786 auto shape_value = GetValueNode(node->input(2))->cast<ValueSequeuePtr>(); in InsertShapeOp() local
2787 MS_EXCEPTION_IF_NULL(shape_value); in InsertShapeOp()
2788 auto shape = shape_value->value(); in InsertShapeOp()
|