| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/ |
| D | gpu_kernel_factory.cc | 61 for (size_t input_index = 0; input_index < attr.GetInputSize(); ++input_index) { in SupportedTypeList() local
66 for (size_t input_index = 0; input_index < attr.GetOutputSize(); ++input_index) { in SupportedTypeList() local
124 for (size_t input_index = 0; input_index < kernel_info->GetInputNum(); input_index++) { in ReducePrecision() local
160 …NativeGpuKernelModFactory::CheckSM(const KernelBuildInfo *kernel_info, const size_t &input_index) { in CheckSM()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | check_consistency.cc | 31 bool CheckFormatForConsistency(const CNodePtr &node, const size_t input_index) { in CheckFormatForConsistency()
70 bool CheckDataTypeForConsistency(const CNodePtr &node, const size_t input_index) { in CheckDataTypeForConsistency()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/ |
| D | npu_executor.h | 39 …void InitInputMappingRelationShip(const std::vector<size_t> &input_index) { input_relationship_ = … in InitInputMappingRelationShip()
|
| D | npu_subgraph.h | 53 …void UpdateInputMappingRelationShip(const std::vector<size_t> &input_index) { input_relationship_ … in UpdateInputMappingRelationShip()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | const_input_to_attr.h | 33 ConstInputToAttrInfoRegister &SetConstInputToAttr(size_t input_index) { in SetConstInputToAttr()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/ |
| D | graph_partitioner.cc | 145 for (size_t input_index = 0; input_index < input_num; ++input_index) { in ExistCycleAfterMerge() local
169 for (size_t input_index = 0; input_index < input_num; ++input_index) { in MergeParentBranchRecursively() local
198 for (size_t input_index = 0; input_index < input_num; ++input_index) { in NodeGrouping() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/ |
| D | add_concat_act_fusion_test.cc | 40 CNodeTptr BuildAdd(const string &name, std::vector<uint32_t> input_index, std::vector<uint32_t> out… in BuildAdd()
79 std::vector<uint32_t> input_index{0, 1}; in BuildGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | unique_impl.cu | 29 cudaError_t CalUnique(const T *input, int num_elements, S *input_index, S *sorted_index, T *output,… in CalUnique()
|
| D | unique_with_pad_impl.cu | 30 cudaError_t CalUniqueWithPad(const T *input, int num_elements, S *input_index, S *sorted_index, T *… in CalUniqueWithPad()
|
| D | euclidean_norm_impl.cu | 33 __inline__ __device__ void EuclideanNormCoreOp(const T *input, size_t input_index, T *output, size_… in EuclideanNormCoreOp()
38 __inline__ __device__ void EuclideanNormCoreOp(const Complex<float> *input, size_t input_index, Com… in EuclideanNormCoreOp()
49 __inline__ __device__ void EuclideanNormCoreOp(const Complex<double> *input, size_t input_index, in EuclideanNormCoreOp()
60 __inline__ __device__ void EuclideanNormHighPrecisionCoreOp(const T *input, size_t input_index, flo… in EuclideanNormHighPrecisionCoreOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/ |
| D | ones_like.cc | 27 constexpr size_t input_index = 0; in InferValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/quant_helper/ |
| D | quant_node_pass.cc | 105 …NodePass::CheckNodeDType(const CNodePtr &cnode, const AnfNodePtr &input_node, size_t input_index) { in CheckNodeDType()
132 …::DoParameterNodeQuant(const CNodePtr &cnode, const ParameterPtr &input_node, size_t input_index) { in DoParameterNodeQuant()
179 …Pass::DoValueNodeQuant(const CNodePtr &cnode, const ValueNodePtr &input_node, size_t input_index) { in DoValueNodeQuant()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ |
| D | config_infos.cc | 120 for (size_t input_index = 0; input_index != inputs_of_str.size(); ++input_index) { in ParseDynamicDims() local
192 for (size_t input_index = 0; input_index != inputs_of_str.size(); ++input_index) { in ParseOptDims() local
301 for (size_t input_index = 0; input_index < input_names.size(); input_index++) { in ReorderByInputNames() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/memory/ |
| D | memory_swap_actor.cc | 33 size_t input_index = IntToSize(input_data->index_); in FetchRealParameters() local
53 size_t input_index = IntToSize(input_data->index_); in UpdateDeviceTensors() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/ |
| D | full_quant_quantizer.cc | 75 … int input_index, const tensor::TensorPtr &tensor_info, bool per_channel) { in QuantWeight()
87 … const PrimitivePtr &primitive, int input_index, bool per_channel) { in DoParameterWeightQuant()
100 … const PrimitivePtr &primitive, int input_index, bool per_channel) { in DoValueNodeWeightQuant()
134 size_t input_index) { in DoParameterNodeQuant()
173 …izer::DoValueNodeQuant(const CNodePtr &cnode, const ValueNodePtr &input_node, size_t input_index) { in DoValueNodeQuant()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | gamma_info.cc | 178 auto input_index = p_edge->next_op_input_index(); in ComputeOpAndPrevEdgeParameterInvolved() local
208 auto input_index = p_edge->next_op_input_index(); in ComputeOpAndPrevEdgeParameterInvolved() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/ |
| D | host_kernel_metadata.cc | 39 for (size_t input_index = 0; input_index < input_num; ++input_index) { in HostMetadataInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/utils/ |
| D | fft_helper.h | 58 int64_t input_index = 0; in ShapeCopy() local
|
| /third_party/mesa3d/src/gallium/targets/teflon/ |
| D | test_executor.cpp | 62 unsigned input_index; in patch_conv2d() local
196 auto input_index = add_op->inputs.data()[0]; in patch_quant_for_add() local
232 unsigned input_index = subgraph->operators[2]->inputs.data()[0]; in add_generate_model() local
260 unsigned input_index; in patch_fully_connected() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_ext_info_handle.cc | 189 bool AicpuExtInfoHandler::UpdateInputShapeAndType(uint32_t input_index, const NotNull<AnfNodePtr> &… in UpdateInputShapeAndType()
208 bool AicpuExtInfoHandler::UpdateInputShapeAndType(uint32_t input_index, const kernel::KernelTensor … in UpdateInputShapeAndType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | refresh_parameter_format.cc | 30 for (size_t input_index = 0; input_index < input_num; input_index++) { in DoRefresh() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/ |
| D | remove_redundant_tensor.cc | 33 for (auto input_index : graph_input_index) { in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | arithmetic_fp32.cc | 131 void ArithmeticCPUKernel::DoBroadcast(void *out_data, int input_index) { in DoBroadcast()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/ |
| D | train_export.h | 40 size_t input_index; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/ |
| D | do_infer.cc | 191 size_t input_index) { in GetInputValueAbs()
209 … const ValueSequencePtr &tuple_value, size_t input_index) { in GetInputTupleValueAbstract()
231 AbstractBasePtr InferOperation::GetAbstractByValue(const ValuePtr &value, size_t input_index, in GetAbstractByValue()
|