| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | output_actor.h | 43 OutputActor(std::string name, size_t loop_count, size_t outputs_num, bool need_loop_count) in OutputActor() argument
47 outputs_num_(outputs_num), in OutputActor()
50 outputs_.resize(outputs_num); in OutputActor()
51 output_nodes_.resize(outputs_num); in OutputActor()
52 device_contexts_.resize(outputs_num); in OutputActor()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_utils.cc | 54 std::vector<size_t> *outputs_num) { in GetOutputsRealNum() argument
65 outputs_num->emplace_back(real_output_num); in GetOutputsRealNum()
67 outputs_num->emplace_back(1); in GetOutputsRealNum()
|
| D | tbe_json_utils.h | 113 std::vector<size_t> *outputs_num);
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/ |
| D | calib_output.cc | 102 size_t outputs_num = outputs.size(); in CompareOutputs() local
103 for (size_t i = 0; i < outputs_num; ++i) { in CompareOutputs()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_x86/benchmark/ |
| D | calib_output.cc | 102 size_t outputs_num = outputs.size(); in CompareOutputs() local
103 for (size_t i = 0; i < outputs_num; ++i) { in CompareOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.h | 64 … const KernelMapPosition &origin_outputs_order, const size_t outputs_num, const std::string &name, in GraphCompilerInfo()
74 outputs_num_(outputs_num), in GraphCompilerInfo()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | params_info.h | 51 size_t outputs_num() const;
|
| D | optimizer_kernel_factory.cc | 50 size_t output_num = params_info.outputs_num(); in Matched()
|
| D | aggregation_kernel_factory.cc | 51 size_t output_num = params_info.outputs_num(); in Matched()
|
| D | params_info.cc | 44 size_t ParamsInfo::outputs_num() const { return outputs_name_type_.size(); } in outputs_num() function in mindspore::fl::server::kernel::ParamsInfo
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 897 auto outputs_num = AnfAlgo::GetOutputTensorNum(output_node); in ConstructOutputs() local
905 } else if (outputs_num != 0) { in ConstructOutputs()
907 (*output_position) += outputs_num; in ConstructOutputs()
918 for (size_t i = 0; i < outputs_num; ++i) { in ConstructOutputs()
927 for (size_t i = 0; i < outputs_num; ++i) { in ConstructOutputs()
963 size_t outputs_num = 0; in ConstructGraphCompilerInfo() local
975 outputs_num = outputs.size(); in ConstructGraphCompilerInfo()
987 … root_graph->parameters(), parser, outputs_order, outputs_num, name, false, in ConstructGraphCompilerInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | split_fission.cc | 62 …id AddNewOutputs(const FuncGraphPtr &func_graph, const AnfNodePtr &new_splitv, int64_t outputs_num, in AddNewOutputs() argument
66 …CreateMultipleOutputsOfAnfNode(func_graph, new_splitv, LongToSize(outputs_num), &new_splitv_output… in AddNewOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 370 size_t outputs_num = AnfAlgo::GetOutputTensorNum(node); in InsertTransOpForOutput() local
371 if (outputs_num == 0) { in InsertTransOpForOutput()
376 if (outputs_num == 1 && (!AnfAlgo::IsTupleOutput(node))) { in InsertTransOpForOutput()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | conv2d.cc | 50 int outputs_num = out_tensors_.size(); in CheckSpecs() local
51 if (outputs_num != OUTPUT_TENSOR_SIZE_1) { in CheckSpecs()
52 MS_LOG(WARNING) << "Conv2D only supports 1 output Tensor but get " << outputs_num; in CheckSpecs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 382 size_t outputs_num = 1; in GetAllOutputWithIndex() local
384 outputs_num = AnfAlgo::GetOutputTensorNum(node); in GetAllOutputWithIndex()
387 for (size_t i = 0; i < outputs_num; ++i) { in GetAllOutputWithIndex()
|