| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | lite_graph.h | 37 void SetOutput(size_t i, const NodePtr &node) { output_->SetInput(i, node); } in SetOutput() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/auto_grad/ |
| D | function_context.h | 95 void SetOutput(const ValuePtr &output) { output_ = output; } in SetOutput() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/acl/ |
| D | acl_model_multi.cc | 200 void AclModelMulti::SetOutput() { in SetOutput() function in mindspore::AclModelMulti
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | op_proto.cc | 172 OpProto &OpProto::SetOutput(const std::string &name, const std::string &tensor_type) { in SetOutput() function in mindspore::transform::OpProto
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/backend/ |
| D | graph_optimizer_test_framework.cc | 137 void ConstructGraph::SetOutput(const AnfNodePtr &node) { in SetOutput() function in mindspore::test::ConstructGraph
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/ |
| D | acl_utils.h | 274 void SetOutput(size_t i, const aclTensorDesc *desc, aclDataBuffer *buffer) { in SetOutput() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | ragged_tensor_to_tensor_cpu_kernel.cc | 318 bool RaggedTensorToTensorCpuKernelMod::SetOutput(const std::vector<kernel::KernelTensor *> &inputs, in SetOutput() function in mindspore::kernel::RaggedTensorToTensorCpuKernelMod
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | quantile.cc | 353 void QuantileCpuKernel::SetOutput(std::vector<int64_t> *out_shape) { in SetOutput() function in aicpu::QuantileCpuKernel
|
| D | ragged_tensor_to_tensor.cc | 549 uint32_t RaggedTensorToTensorCpuKernel::SetOutput(CpuKernelContext &ctx, const vector<INDEX_TYPE> &… in SetOutput() function in aicpu::RaggedTensorToTensorCpuKernel
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/ |
| D | ir_grad.cc | 732 void IrGrad::SetOutput(const tensor::BaseTensorPtrList &weights, const std::vector<size_t> &grad_po… in SetOutput() function in mindspore::pynative::autograd::IrGrad
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ |
| D | ge_graph_executor.cc | 1162 void SetOutput(GeDeviceResManager *res_manager, GeTensor *ge_output, const AnfNodePtr &output_node,… in SetOutput() function
|