Home
last modified time | relevance | path

Searched defs:output_idx (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc406 for (auto output_idx : op->outputs) { in ConvertOpQuantParams() local
468 … auto output_idx = output_node < 0 ? output_node + tflite_subgraph->tensors.size() : output_node; in ConvertGraphOutputs() local
507 int output_idx = tflite_subgraph->outputs.front() < 0 in ConvertGraphOutputs() local
529 auto output_idx = tflite_subgraph->outputs; in ConvertGraphOutputs() local
743 int output_idx = in ConvertOutputTensor() local
757 for (auto output_idx : op->outputs) { in ConvertOutputTensor() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_swap_manager.cc41 for (size_t output_idx = 0; output_idx < output_num; ++output_idx) { in Init() local
248 auto &output_idx = kernel_with_index.second; in SaveUserKernelTopoOrder() local
283 size_t output_idx = tensor.output_idx_; in AddSwapInfo() local
488 void MemSwapManager::AddKernelSwapPerform(const AnfNodePtr &kernel, size_t output_idx, in AddKernelSwapPerform()
619 void MemSwapManager::AddKernelHostAddrIsDirty(const AnfNodePtr &kernel, size_t output_idx, bool dir… in AddKernelHostAddrIsDirty()
717 size_t output_idx = item.first; in DumpUserNodes() local
Dmem_reuse.cc250 KernelRefCountPtr MemReuseUtil::GetRef(const AnfNodePtr &node, size_t output_idx) { in GetRef()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc733 std::string AnfRuntimeAlgorithm::GetOutputFormat(const AnfNodePtr &node, size_t output_idx) { in GetOutputFormat()
808 size_t output_idx) { in GetOutputInferShape()
845 …ector<size_t> AnfRuntimeAlgorithm::GetOutputInferShape(const AnfNodePtr &node, size_t output_idx) { in GetOutputInferShape()
856 const size_t output_idx, in GetOutputDeviceShapeForTbeBuild()
876 …ctor<size_t> AnfRuntimeAlgorithm::GetOutputDeviceShape(const AnfNodePtr &node, size_t output_idx) { in GetOutputDeviceShape()
923 std::string AnfRuntimeAlgorithm::GetOutputReshapeType(const AnfNodePtr &node, size_t output_idx) { in GetOutputReshapeType()
943 TypeId AnfRuntimeAlgorithm::GetOutputInferDataType(const TypePtr &type, size_t output_idx) { in GetOutputInferDataType()
967 TypeId AnfRuntimeAlgorithm::GetOutputInferDataType(const AnfNodePtr &node, size_t output_idx) { in GetOutputInferDataType()
977 TypeId AnfRuntimeAlgorithm::GetOutputDeviceDataType(const AnfNodePtr &node, size_t output_idx) { in GetOutputDeviceDataType()
1029 const DeviceAddress *AnfRuntimeAlgorithm::GetOutputAddr(const AnfNodePtr &node, size_t output_idx, in GetOutputAddr()
[all …]
Dkernel_graph.cc665 AnfNodePtr KernelGraph::CreatTupleGetItemNode(const AnfNodePtr &node, size_t output_idx) { in CreatTupleGetItemNode()
1041 …elGraph::AddInternalOutput(const AnfNodePtr &front_node, const AnfNodePtr &node, size_t output_idx, in AddInternalOutput()
1055 void KernelGraph::AddInternalOutputTensor(const AnfNodePtr &node, size_t output_idx, const tensor::… in AddInternalOutputTensor()
1062 tensor::TensorPtr KernelGraph::GetInternalOutputTensor(const AnfNodePtr &node, size_t output_idx) { in GetInternalOutputTensor()
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Doptimizer_kernel_factory.cc58 size_t output_idx = output_name_to_idx.at(name); in Matched() local
Daggregation_kernel_factory.cc60 size_t output_idx = output_name_to_idx.at(name); in Matched() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Done_hot_impl.cu31 size_t output_idx = left_idx * depth * right_dim_size + d_idx * right_dim_size + right_idx; in OneHotKernel() local
Dslice_copy_impl.cu56 …size_t output_idx = (i * stride[0] + begins[0]) * o_offset[0] + (j * stride[1] + begins[1]) * o_of… in CopySlicesKernel() local
Dtranspose_impl_opt.cu155 int output_idx = output_block_start_idx + shm_col_id * output_dims[2] + shm_row_id; in Swap3DTensorLast2DimKernel_shared() local
/third_party/boost/boost/compute/algorithm/detail/
Dfind_extrema_on_cpu.hpp63 buffer output_idx( in find_extrema_on_cpu() local
/third_party/boost/libs/compute/include/boost/compute/algorithm/detail/
Dfind_extrema_on_cpu.hpp63 buffer output_idx( in find_extrema_on_cpu() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dmerge_cast_to_op.cc189 …GetPriorOp(const AnfNodePtr &x_node, CNodePtr *prior_op, bool *single_output, size_t *output_idx) { in GetPriorOp()
228 size_t output_idx = 0; in MergeCastToPriorOp() local
Dinsert_transpose_for_basiclstm_op.cc63 for (size_t output_idx = 0; output_idx < out_num; output_idx++) { in Insert() local
Dinsert_cast.cc48 for (size_t output_idx = 0; output_idx < out_num; ++output_idx) { in InsertCastForMultipleOutput() local
/third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dsubgraph_node_pass.cc45 for (auto &output_idx : node->outputIndex) { in GetSubgraphAllTensorIndices() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dactor_common.cc180 void UpdateRefCount(const AnfNodePtr &node, size_t output_idx, bool is_max_ref_count) { in UpdateRefCount()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc333 auto output_idx = GetValue<int64_t>(GetValueNode(input2)); in GetFusionScopeOutputNodeList() local
377 auto output_idx = GetValue<int64_t>(GetValueNode(input2)); in SetFusionOpRefInfos() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Ddata_utils.cc780 std::shared_ptr<Tensor> *output_idx, std::shared_ptr<Tensor> *output_cnt) { in UniqueHelper()
816 std::shared_ptr<Tensor> *output_idx, std::shared_ptr<Tensor> *output_cnt) { in Unique()
/third_party/gstreamer/gstplugins_bad/sys/d3d11/
Dgstd3d11screencapturedevice.cpp402 for (UINT output_idx = 0;; output_idx++) { in gst_d3d11_screen_capture_device_provider_probe() local
Dgstd3d11screencapture.cpp1937 for (UINT output_idx = 0;; output_idx++) { in gst_d3d11_screen_capture_find_output_for_monitor() local
1981 for (UINT output_idx = 0;; output_idx++) { in gst_d3d11_screen_capture_find_primary_monitor() local
2033 for (UINT output_idx = 0;; output_idx++) { in gst_d3d11_screen_capture_find_nth_monitor() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.cc241 int output_idx = PosInInputEnterNodes(enter_node); in UpdateExitNodeUser() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc486 … CreatTupleGetItemNode(const FuncGraphPtr &func_graph, const AnfNodePtr &node, size_t output_idx) { in CreatTupleGetItemNode()
1010 auto output_idx = LongToSize(GetValue<int64_t>(GetValueNode(getitem_input2))); in GetNodeOutputUsedNum() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dtranspose_fp32.c195 int output_idx = 0; in TransposeDimsFp32() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc169 for (size_t output_idx = 0; output_idx < out_num; ++output_idx) { in InsertTransOpForMultipleOutput() local

12