Home
last modified time | relevance | path

Searched refs:inputs_ptr (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel_factory.cc62 auto inputs_ptr = op_info->inputs_ptr(); in SetKernelAttrs() local
64 if (inputs_ptr.empty()) { in SetKernelAttrs()
67 auto first_input_dtypes = inputs_ptr[0]->dtypes(); in SetKernelAttrs()
68 auto input_formats = inputs_ptr[0]->formats(); in SetKernelAttrs()
73 for (size_t j = 1; j < inputs_ptr.size(); j++) { in SetKernelAttrs()
74 auto input_dtypes = inputs_ptr[j]->dtypes(); in SetKernelAttrs()
75 input_formats = inputs_ptr[j]->formats(); in SetKernelAttrs()
/third_party/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_fs_linear_llvm.c165 LLVMValueRef inputs_ptr; in llvm_fragment_body() local
168 inputs_ptr = inputs_ptrs[i]; in llvm_fragment_body()
170 input = lp_build_pointer_get(builder, inputs_ptr, sampler->counter); in llvm_fragment_body()
382 LLVMValueRef inputs_ptr; in llvmpipe_fs_variant_linear_llvm() local
400 inputs_ptr = LLVMBuildCall(builder, fetch_ptr, &elem, 1, ""); in llvmpipe_fs_variant_linear_llvm()
401 assert(LLVMGetTypeKind(LLVMTypeOf(inputs_ptr)) == LLVMPointerTypeKind); in llvmpipe_fs_variant_linear_llvm()
404 lp_add_function_attr(inputs_ptr, -1, LP_FUNC_ATTR_READONLY); in llvmpipe_fs_variant_linear_llvm()
405 lp_add_function_attr(inputs_ptr, -1, LP_FUNC_ATTR_NOUNWIND); in llvmpipe_fs_variant_linear_llvm()
407 lp_build_name(inputs_ptr, "input%u_ptr", attrib); in llvmpipe_fs_variant_linear_llvm()
409 inputs_ptrs[attrib] = inputs_ptr; in llvmpipe_fs_variant_linear_llvm()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dsingle_tbe_json_creator.cc76 auto inputs_ptr = op_info_ptr->inputs_ptr(); in GenDataJson() local
78 for (size_t i = 0; i < inputs_ptr.size(); i++) { in GenDataJson()
80 auto input_ptr = inputs_ptr[i]; in GenDataJson()
106 std::vector<OpIOInfoPtr> inputs_ptr = op_info_ptr->inputs_ptr(); in GenInputsJson() local
108 if (inputs_ptr.empty()) { in GenInputsJson()
114 if (!TbeJsonUtils::GetInputsRealNum(anf_node, inputs_ptr, &inputs_tensor_num)) { in GenInputsJson()
137 if (!AssignInputsJson(anf_node, inputs_desc, inputs_tensor_num, inputs_ptr, &inputs_list)) { in GenInputsJson()
189 const std::vector<OpIOInfoPtr> &inputs_ptr, in AssignInputsJson() argument
196 auto input_ptr = inputs_ptr[i]; in AssignInputsJson()
Dtbe_json_utils.cc26 …JsonUtils::GetInputsRealNum(const AnfNodePtr &anf_node, const std::vector<OpIOInfoPtr> &inputs_ptr, in GetInputsRealNum() argument
37 for (const auto &input_ptr : inputs_ptr) { in GetInputsRealNum()
Dsingle_tbe_json_creator.h36 … const std::vector<size_t> &inputs_tensor_num, const std::vector<OpIOInfoPtr> &inputs_ptr,
Dtbe_json_utils.h110 …tatic bool GetInputsRealNum(const AnfNodePtr &anf_node, const std::vector<OpIOInfoPtr> &inputs_ptr,
Dfusion_tbe_json_creator.cc202 if (!TbeJsonUtils::GetInputsRealNum(anf_node, op_info->inputs_ptr(), &inputs_tensor_num)) { in GenInputsJson()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc96 const auto inputs_info = op_info.inputs_ptr(); in GetCommonPatternKernelInfo()
144 if (op_info.inputs_ptr().size() != 1) { in GetAgnosticPatternKernelInfo()
155 input_item.assign(op_info.inputs_ptr().size(), format); in GetAgnosticPatternKernelInfo()
479 if (op_info.inputs_ptr().size() != support_format.input_format[0].size() || in CreateNewOpInfo()
481 …ION) << "BroadCast input/output size not match, op info input size:" << op_info.inputs_ptr().size() in CreateNewOpInfo()
489 for (size_t i = 0; i < op_info.inputs_ptr().size(); ++i) { in CreateNewOpInfo()
490 auto input = op_info.inputs_ptr().at(i); in CreateNewOpInfo()
549 …if (op_info.inputs_ptr().size() != inputs.size() || op_info.outputs_ptr().size() != outputs.size()… in CreateNewOpInfo()
556 for (size_t i = 0; i < op_info.inputs_ptr().size(); ++i) { in CreateNewOpInfo()
558 …CreateNewOpIOInfo(*op_info.inputs_ptr().at(i), inputs.at(i).dtypes, inputs.at(i).formats, input_ne… in CreateNewOpInfo()
/third_party/mindspore/mindspore/ccsrc/vm/
Dsegment_runner.cc73 …r RefSubGraphNode(const FuncGraphPtr &fg, const AnfNodePtr &node, AnfNodePtrList *const inputs_ptr, in RefSubGraphNode() argument
76 MS_EXCEPTION_IF_NULL(inputs_ptr); in RefSubGraphNode()
79 auto &inputs = *inputs_ptr; in RefSubGraphNode()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dcostmodel.h82 …: strategy_ptr(std::move(strategy)), inputs_ptr(std::move(inputs_)), outputs_ptr(std::move(outputs… in StrategyWithCost()
89 inputs_ptr(swc.inputs_ptr), in StrategyWithCost()
95 std::vector<TensorInfo> inputs_ptr; variable
Dedge_costmodel.cc41 next_op_input_.emplace_back(std::make_pair(swc->strategy_ptr, swc->inputs_ptr)); in InitEdgeCost()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/
Dop_tiling_adapter.cc238 auto inputs_ptr = op_info_ptr->inputs_ptr(); in InitOpIoName() local
244 for (const auto &item : inputs_ptr) { in InitOpIoName()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc174 auto inputs_ptr = op_info->inputs_ptr(); in CreateInputDescJson() local
175 if (inputs_ptr.empty()) { in CreateInputDescJson()
183 for (size_t i = 0; i < inputs_ptr.size(); i++) { in CreateInputDescJson()
184 auto input_ptr = inputs_ptr[i]; in CreateInputDescJson()
309 auto inputs = op_info->inputs_ptr(); in CreateAttrDescJson()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/
Dopinfo.h117 for (const auto &input : opinfo.inputs_ptr()) { in OpInfo()
138 std::vector<std::shared_ptr<OpIOInfo>> inputs_ptr() const { return inputs_ptr_; } in inputs_ptr() function
Doplib.cc310 …if (!DecodeDtypeFormat(dtype_format, op_io, op_info->inputs_ptr().size() + op_info->outputs_ptr().… in DecodeInputOutput()
388 const auto &input_infos = op_info->inputs_ptr(); in GetRefInfo()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreshape_info.cc452 pre_out_tensor_infos = pre_stra_cost->inputs_ptr; in GenetateStrategyCosts()
484 std::vector<TensorInfo> next_in_tensor_infos = next_stra_cost->inputs_ptr; in GenetateStrategyCosts()
Doperator_info.cc1529 auto mem_cost = operator_cost()->GetMemoryCost(swc->inputs_ptr, swc->outputs_ptr); in CalculateMemoryCost()
1544 auto mem_cost = operator_cost()->GetMemoryCostForInference(swc->inputs_ptr, swc->outputs_ptr); in CalculateMemoryCostForInference()
1552 double parameter_mem_cost = ListProduct(swc->inputs_ptr[input_index].slice_shape()) * in CorrectMemoryCost()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc429 std::vector<std::shared_ptr<OpIOInfo>> inputs_ptr = op_info->inputs_ptr(); in GenTbeInputsJson() local
430 if (inputs_ptr.empty()) { in GenTbeInputsJson()
434 auto op_info_input_num = inputs_ptr.size(); in GenTbeInputsJson()
440 std::shared_ptr<OpIOInfo> input_ptr = inputs_ptr[i]; in GenTbeInputsJson()
1340 if (op_info->inputs_ptr().size() < (node_inputs_size - 1)) { in GetOptionalInput()
1343 return (op_info->inputs_ptr().size() + 1 - node_inputs_size); in GetOptionalInput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.h61 bool ConvertNonscalarTensorToParameter(const FuncGraphPtr &fg, AnfNodePtrList *inputs_ptr);
Dgraph_kernel_helper.cc189 bool ConvertNonscalarTensorToParameter(const FuncGraphPtr &fg, AnfNodePtrList *inputs_ptr) { in ConvertNonscalarTensorToParameter() argument
190 MS_EXCEPTION_IF_NULL(inputs_ptr); in ConvertNonscalarTensorToParameter()
229 auto &inputs = *inputs_ptr; in ConvertNonscalarTensorToParameter()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dkernel_select_cpu.cc260 const auto &input_io_info = op_info->inputs_ptr(); in IsDynamicParamKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc435 std::vector<std::shared_ptr<OpIOInfo>> inputs = op_info_ptr->inputs_ptr(); in ParseMetadata()