Home
last modified time | relevance | path

Searched defs:tensor_out (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/
Dtensor_proto.cc75 std::shared_ptr<Tensor> tensor_out; in PbToTensor() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_cost.cc259 int64_t tensor_out = output_tensor_h * output_tensor_w * output_tensor_n * output_tensor_c; in GetOptimalStr() local
308 …int64_t tensor_out = static_cast<int64_t>(node.tensor_parm.tensor_shape.shape_h * node.tensor_parm… in GetMinCostIn() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dbounding_box.cc110 TensorPtr *tensor_out) { in CreateTensorFromBoundingBoxList()
/third_party/mindspore/mindspore/lite/src/
Dlite_kernel.h266 auto tensor_out = mindspore::MSTensor(impl); in set_out_tensor() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_others.cc429 auto tensor_out = std::make_shared<AbstractTensor>(tensor_in->element(), in InferImplAllSwap() local