| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_add_layernorm.py | 57 def test_add_layer_norm(tensor_type): argument
92 def test_add_layer_norm_dynamic_shape(tensor_type): argument
|
| D | test_quant_v2.py | 27 def generate_random_input(shape, dtype, tensor_type): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | op_proto.cc | 165 OpProto &OpProto::SetInput(const std::string &name, const std::string &tensor_type, bool is_optiona… in SetInput()
172 OpProto &OpProto::SetOutput(const std::string &name, const std::string &tensor_type) { in SetOutput()
185 OpProto &OpProto::DefineDataType(const std::string &name, const std::string &tensor_type) { in DefineDataType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/utils/ |
| D | tensor_construct_utils.cc | 73 const auto &tensor_type = tensor->data_type(); in CreateOnesTensor() local
97 auto tensor_type = type->cast<TensorTypePtr>(); in ExtractTypeId() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/proto/ |
| D | tensor.proto | 29 int32 tensor_type = 2; field
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/ |
| D | print.proto | 25 required string tensor_type = 2; field
|
| D | checkpoint.proto | 35 required string tensor_type = 2; field
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | fast_gelu.cc | 31 auto tensor_type = input_args[0]->GetType()->cast<TensorTypePtr>(); in InferType() local
|
| D | fast_gelu_grad.cc | 31 auto tensor_type = input_args[0]->GetType()->cast<TensorTypePtr>(); in InferType() local
|
| D | sum_ext.cc | 32 auto tensor_type = input_args[kIndex0]->GetType()->cast<TensorTypePtr>(); in InferType() local
|
| D | non_zero.cc | 52 auto tensor_type = input_args[kInputIndex0]->GetType(); in CheckValidation() local
|
| D | apply_rotary_pos_emb.cc | 51 auto tensor_type = cos_type->cast<TensorTypePtr>(); in InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | multi_margin_loss_grad.cc | 55 auto tensor_type = input_args[kInputIndex3]->GetType()->cast<TensorTypePtr>(); in MultiMarginLossGradInferType() local
88 auto tensor_type = input_args[kInputIndex3]->GetType()->cast<TensorTypePtr>(); in MultiMarginLossGradInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | multi_margin_loss.cc | 58 auto tensor_type = input_args[kInputIndex2]->GetType()->cast<TensorTypePtr>(); in MultiMarginLossInferType() local
104 auto tensor_type = input_args[kInputIndex2]->GetType()->cast<TensorTypePtr>(); in MultiMarginLossInferShape() local
|
| D | tensor_to_scalar.cc | 74 auto tensor_type = x_type->cast<TensorTypePtr>(); in InferType() local
|
| D | fill.cc | 72 const auto &tensor_type = tensor->data_type(); in CreateValuedTensor() local
98 const auto &tensor_type = tensor->data_type(); in CreateComplexTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | dropout_ext.cc | 44 auto tensor_type = op->input_abs()[kIndex0]->GetType()->cast<TensorTypePtr>(); in DropoutExtAscendCustomize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/symbolic_shape/ |
| D | symbol.cc | 76 auto tensor_type = type->cast_ptr<TensorType>(); in ToValueOf() local
|
| D | int_symbol.cc | 31 auto tensor_type = type->cast_ptr<TensorType>(); in ToValueOf() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_tensor.h | 42 TensorType tensor_type = kFloat32; // default as float. member
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | cast_tests.cc | 63 auto tensor_type = lite::Category::CONST_TENSOR; in TEST_F() local
172 auto tensor_type = lite::Category::CONST_TENSOR; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | float32_redistribution.cc | 60 auto tensor_type = node->abstract()->BuildType(); in Float32Redistribution() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/ |
| D | transpose_optimizer.cc | 76 auto tensor_type = tensor->data_type_c(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/ |
| D | unstack_ext.cc | 82 auto tensor_type = input_type->cast<TensorTypePtr>(); in InferAbstract() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | trans_depend_value_to_int32.cc | 62 TensorTypePtr tensor_type = std::make_shared<TensorType>(kInt32); in TransValueToInt32() local
|