Home
last modified time | relevance | path

Searched defs:tensor_ (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Ddevice_tensor_ascend910b.h84 void *tensor_; // aclTensor which point to device_address_ variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/
Dlayer_input.h62 nvinfer1::ITensor *tensor_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/auto_grad/
Dfunction_node.h193 py::object tensor_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/
Dforward_task.h157 tensor::TensorPtr tensor_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/
Daclnn_kernel_mod.h124 ~EmptyKernelTensor() { delete tensor_; } in ~EmptyKernelTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dtransform_util.cc53 mutable MeTensorPtr tensor_; member in mindspore::transform::__anon593569c90111::MsTensorRel
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_cross_cpu_kernel.cc90 std::vector<T> tensor_; member in mindspore::kernel::DenseTensorColumn
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.h378 TensorPtr tensor_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_cross.cc123 Tensor *tensor_; member in aicpu::DenseTensorColumn
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/
Dparse_example_op.cc555 std::shared_ptr<Tensor> tensor_ = nullptr; member in mindspore::dataset::TensorVector