Home
last modified time | relevance | path

Searched defs:input_index (Results 1 – 25 of 93) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel_factory.cc64 for (size_t input_index = 0; input_index < attr.GetInputSize(); ++input_index) { in SupportedTypeList() local
69 for (size_t input_index = 0; input_index < attr.GetOutputSize(); ++input_index) { in SupportedTypeList() local
91 for (size_t input_index = 0; input_index < kernel_info->GetInputNum(); input_index++) { in ReducePrecision() local
112 void GpuKernelFactory::CheckSM(const KernelBuildInfo *kernel_info, const size_t &input_index) { in CheckSM()
147 for (size_t input_index = 0; input_index < kernel_info->GetInputNum(); input_index++) { in GpuKernelAttrCheck() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dcheck_consistency.cc30 bool CheckFormatForConsistency(const CNodePtr &node, const size_t input_index) { in CheckFormatForConsistency()
69 bool CheckDataTypeForConsistency(const CNodePtr &node, const size_t input_index) { in CheckDataTypeForConsistency()
Ddeal_ref_and_split_unsupported_transdata.cc101 size_t input_index, in AddAdditionalToRefOutput()
192 auto input_index = ref_infos.at(output_index); in DealRefForMultipleOutput() local
Dconvert_cast_format.cc46 for (size_t input_index = 0; input_index < input_num; ++input_index) { in Process() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.cc61 for (size_t input_index = 0; input_index < kernel_build_info.GetInputNum(); ++input_index) { in MatchInferOutputDataType() local
145 for (size_t input_index = 0; input_index < input_num; ++input_index) { in UpdateCurMatchCounts() local
260 for (size_t input_index = 0; input_index < kernel_build_info->GetInputNum(); ++input_index) { in TagRaiseReduce() local
361 size_t input_index, bool force_fresh = false) { in SetWeightFormat()
437 for (size_t input_index = 0; input_index < input_num; ++input_index) { in SetTensorDeviceInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dunique_impl.cu30 int CalUnique(const T *input, int num_elements, S *input_index, S *sorted_index, T *output, S *inde… in CalUnique()
Dunsorted_segment_sum.cu23 …for (int input_index = blockIdx.x * blockDim.x + threadIdx.x; input_index < input_dim0 * input_dim… in UnsortedSegmentSum() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.cc139 for (size_t input_index = 0; input_index < input_num; ++input_index) { in ExistCycleAfterMerge() local
163 for (size_t input_index = 0; input_index < input_num; ++input_index) { in MergeParentBranchRecursively() local
192 for (size_t input_index = 0; input_index < input_num; ++input_index) { in NodeGrouping() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/
Dattention_quant_type_determiner.cc32 auto input_index = node->inputIndex; in DetermineQuantWeight() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dkernel_select_cpu.cc36 bool IsInputNotCNode(const CNodePtr &kernel_node, size_t input_index) { in IsInputNotCNode()
47 for (auto &input_index : input_not_cnode_indexes) { in UpdatePrevNotCNodeFormatDtype() local
82 for (size_t input_index = 0; input_index < input_num; ++input_index) { in GetInputDtypes() local
96 for (size_t input_index = 0; input_index < input_num; ++input_index) { in GetInputFormat() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc44 for (size_t input_index = 0; input_index < input_num; ++input_index) { in HostMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Drefresh_parameter_format.cc28 for (size_t input_index = 0; input_index < input_num; input_index++) { in DoRefresh() local
Dparameter_and_transop_fusion.cc103 for (size_t input_index = 0; input_index < input_num; input_index++) { in Run() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dgraph_desc_reporter.cc38 auto input_index = input_node_with_index.second; in ReportData() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc59 for (size_t input_index = 0; input_index < input_num; ++input_index) { in AicpuMetadataInfoForSpecialNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dconst_input_to_attr_registry.h31 ConstInputToAttrInfoRegister &SetConstInputToAttr(size_t input_index) { in SetConstInputToAttr()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_split_parser.cc42 int input_index; in Parse() local
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Doptimizer_kernel.h71 for (size_t input_index = 0; input_index < input_num; ++input_index) { in InitServerKernelInputOutputSize() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dreplace_node_by_proxy.cc34 for (size_t input_index = 0; input_index < input_num; ++input_index) { in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc90 for (size_t input_index = 0; input_index < input_num; ++input_index) { in HcclMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadd_relu_grad_v2_fusion.cc38 for (size_t input_index = 0; input_index < input_num; ++input_index) { in GenerateKernelBuildInfo() local
Dadd_relu_v2_fusion.cc38 for (size_t input_index = 0; input_index < input_num; ++input_index) { in GenerateKernelBuildInfo() local
Dmatmul_biasadd_fusion.cc38 for (size_t input_index = 0; input_index < input_num; ++input_index) { in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dunique_gpu_kernel.h42 S *input_index = GetDeviceAddress<S>(workspace, 0); in Launch() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dquant_cast.cc43 STATUS InsertCastNode(const FuncGraphPtr &graph, const CNodePtr &cnode, size_t input_index, bool is… in InsertCastNode()

1234