Home
last modified time | relevance | path

Searched refs:GetPrevNodeOutputInferDataType (Results 1 – 25 of 50) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dinput_to_output_registry.cc25 return !(AnfAlgo::GetPrevNodeOutputInferDataType(node, 0) != kNumberTypeFloat32); in ApplyRMSPropPreCheck()
29 TypeId data_type = AnfAlgo::GetPrevNodeOutputInferDataType(node, 0); in FusedMulApplyMomentumPreCheck()
34 return !(AnfAlgo::GetPrevNodeOutputInferDataType(node, 0) != kNumberTypeFloat32); in SparseApplyRMSPropPreCheck()
38 TypeId data_type = AnfAlgo::GetPrevNodeOutputInferDataType(node, 0); in ApplyAdagradV2PreCheck()
43 TypeId data_type = AnfAlgo::GetPrevNodeOutputInferDataType(node, 0); in ApplyKerasMomentumPreCheck()
48 return !(AnfAlgo::GetPrevNodeOutputInferDataType(node, 0) != kNumberTypeFloat32); in SparseApplyFtrlPreCheck()
52 return !(AnfAlgo::GetPrevNodeOutputInferDataType(node, 0) != kNumberTypeFloat32); in SparseApplyFtrlV2PreCheck()
56 return !(AnfAlgo::GetPrevNodeOutputInferDataType(node, 0) != kNumberTypeFloat32); in SparseApplyAdagradV2PreCheck()
60 return !(AnfAlgo::GetPrevNodeOutputInferDataType(node, 0) != kNumberTypeFloat32); in SparseApplyAdadeltaPreCheck()
Dderelu_fusion.cc61 auto input_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(relu, 0); in CreateReluV2()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dreplace_momentum_cast_fusion.cc37 auto src = AnfAlgo::GetPrevNodeOutputInferDataType(grad_cast, 0); in Process()
54 outputs_type[kGradIndex] = AnfAlgo::GetPrevNodeOutputInferDataType(grad_cast, 0); in Process()
Dinsert_format_transform_op.cc65 auto input_type = AnfAlgo::GetPrevNodeOutputInferDataType(node, 0); in SetTransposeOpBuildInfo()
102 auto transpose_type = {AnfAlgo::GetPrevNodeOutputInferDataType(used_node, used_node_index)}; in InsertTransposeOp()
Dadd_relu_v2_fusion.cc39 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
Dmatmul_biasadd_fusion.cc39 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
Dadd_relu_grad_v2_fusion.cc39 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
Dinsert_cast_gpu.cc81 auto inferType = AnfAlgo::GetPrevNodeOutputInferDataType(node, i); in Run()
Dreduce_precision_fusion.cc79 auto inferType = AnfAlgo::GetPrevNodeOutputInferDataType(node, i); in Run()
Dcombine_cast_fusion.cc43 auto src = AnfAlgo::GetPrevNodeOutputInferDataType(cast_node, 0); in GetDealList()
Drelu_v2_pass.cc52 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dbn_grad_unify_mindir.cc49 AnfAlgo::GetPrevNodeOutputInferDataType(bn_grad_node, 3), in CreateNewBatchNormGrad()
50 AnfAlgo::GetPrevNodeOutputInferDataType(bn_grad_node, 4)}; in CreateNewBatchNormGrad()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dunsorted_segment_sum_fission.cc36 …AnfAlgo::SetOutputInferTypeAndShape({AnfAlgo::GetPrevNodeOutputInferDataType(origin_node, 0)}, {sh… in CreatePadding()
116 auto input_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(origin_node, 0); in Process()
Dgather_v2_ds_fission.cc55 auto type_id = AnfAlgo::GetPrevNodeOutputInferDataType(origin_node, 0); in CreatePad()
166 auto input_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(origin_node, 0); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/
Dinsert_format_transform_op.cc65 auto transpose_type = {AnfAlgo::GetPrevNodeOutputInferDataType(used_node, used_node_index)}; in InsertTransposeOp()
79 auto input_type = AnfAlgo::GetPrevNodeOutputInferDataType(node, 0); in SetTransposeOpBuildInfo()
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Doptimizer_kernel_factory.cc42 TypeId kernel_node_input_type = AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_idx); in Matched()
Daggregation_kernel_factory.cc43 TypeId kernel_node_input_type = AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_idx); in Matched()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc46 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); in HostMetadataInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc61 … (void)inputs_type.emplace_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); in AicpuMetadataInfoForSpecialNodes()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Drt_kernel_info.cc75 input_types.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, i)); in GetRtKelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_pad_for_nms_with_mask.cc67 auto origin_type = AnfAlgo::GetPrevNodeOutputInferDataType(cnode, input_idx); in Process()
Dsplit_inputs_for_reduce_scatter.cc34 std::vector<TypeId> dtypes(rank_size, AnfAlgo::GetPrevNodeOutputInferDataType(node, i)); in InsertSplitForInput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_transpose_for_basiclstm_op.cc46 auto origin_type = AnfAlgo::GetPrevNodeOutputInferDataType(cnode, 1); in Insert()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dresize_nearest_neighbor_grad_cpu_kernel.cc36 dtype_ = AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, 0); in InitKernel()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc432 TEST_F(AnfRuntimeAlgorithmTest, GetPrevNodeOutputInferDataType) { in TEST_F() argument
443 EXPECT_EQ(AnfAlgo::GetPrevNodeOutputInferDataType(add, 0), kFloat32->type_id()); in TEST_F()
444 EXPECT_THROW(AnfAlgo::GetPrevNodeOutputInferDataType(add, 1), std::runtime_error); in TEST_F()
445 EXPECT_THROW(AnfAlgo::GetPrevNodeOutputInferDataType(nullptr, 0), std::runtime_error); in TEST_F()
448 EXPECT_THROW(AnfAlgo::GetPrevNodeOutputInferDataType(parameter_node, 0), std::runtime_error); in TEST_F()

12