| /third_party/mindspore/mindspore/core/ops/ |
| D | ceil.cc | 36 auto infer_type = input_args[0]->BuildType(); in CeilInfer() local
|
| D | asin.cc | 39 …auto infer_type = CheckAndConvertUtils::CheckTensorTypeValid("x_dtype", dtype, valid_types, prim_n… in AsinInfer() local
|
| D | round.cc | 32 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| D | ones_like.cc | 40 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| D | dropout.cc | 55 auto infer_type = in DropoutInfer() local
|
| D | zeros_like.cc | 38 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| D | custom_normalize.cc | 41 auto infer_type = input_args[0]->BuildType(); in CustomNormalizeInferType() local
|
| D | roll.cc | 46 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| D | flatten.cc | 43 auto infer_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); in InferType() local
|
| D | erfinv.cc | 50 auto infer_type = input_args[0]->BuildType(); in ErfinvInferType() local
|
| D | reverse_v2.cc | 37 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| D | stack.cc | 56 auto infer_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); in StackInfer() local
|
| D | cdist.cc | 63 auto infer_type = InferType(primitive, input_args); in CdistInfer() local
|
| D | ctc_loss_v2_grad.cc | 70 auto infer_type = CTCLossV2GradInferType(primitive, input_args); in CTCLossV2GradInfer() local
|
| D | concat.cc | 72 auto infer_type = CheckAndConvertUtils::CheckTensorTypeSame(types, all_types, prim_name); in ConcatInfer() local
|
| D | skip_gram.cc | 38 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| D | binary_cross_entropy.cc | 63 auto infer_type = CheckAndConvertUtils::CheckTensorTypeSame(types, valid_types, prim->name()); in BinaryCrossEntroyInferType() local
|
| D | one_hot.cc | 91 auto infer_type = OneHotInferType(primitive, input_args); in OneHotInfer() local
|
| D | split_v.cc | 86 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| D | mfcc.cc | 45 auto infer_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); in InferType() local
|
| D | batch_to_space_nd.cc | 60 auto infer_type = input_args[0]->BuildType(); in InferType() local
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | cdist_grad.cc | 66 auto infer_type = InferType(primitive, input_args); in CdistGradInfer() local
|
| D | binary_cross_entropy_grad.cc | 45 auto infer_type = CheckAndConvertUtils::CheckTensorTypeSame(types, valid_types, prim->name()); in BinaryCrossEntroyGradInferType() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_cast_cpu.cc | 100 auto infer_type = AnfAlgo::GetOutputInferDataType(cnode, i); in InsertCastForGraphOutput() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 269 auto infer_type = AnfAlgo::GetOutputInferDataType(input, 0); in NewTransOpNode() local
427 const auto infer_type = AnfAlgo::GetOutputInferDataType(prev_node.first, prev_node.second); in InsertCastForInput() local
|