Searched refs:tensor_out (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/ |
| D | tensor_proto.cc | 75 std::shared_ptr<Tensor> tensor_out; in PbToTensor() local
78 tensor_pb->data().size(), &tensor_out)); in PbToTensor()
79 *tensor = std::move(tensor_out); in PbToTensor()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_cost.cc | 259 int64_t tensor_out = output_tensor_h * output_tensor_w * output_tensor_n * output_tensor_c; in GetOptimalStr() local
288 cost_op.push_back(StrDimQ(tensor_out) + CostRedis(node, node_name_to_strategy, in GetOptimalStr()
308 …int64_t tensor_out = static_cast<int64_t>(node.tensor_parm.tensor_shape.shape_h * node.tensor_parm… in GetMinCostIn() local
318 cost_in.push_back(StrDimDI(tensor_in, tensor_out)); in GetMinCostIn()
319 cost_in.push_back(StrDimDJ(tensor_in, tensor_out)); in GetMinCostIn()
320 cost_in.push_back(StrDimQ(tensor_out)); in GetMinCostIn()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/ |
| D | bounding_box.h | 96 TensorPtr *tensor_out);
|
| D | bounding_box.cc | 110 TensorPtr *tensor_out) { in CreateTensorFromBoundingBoxList() argument
117 …OK(Tensor::CreateFromVector(bboxes_for_tensor, TensorShape{num_of_boxes, kNumOfCols}, tensor_out)); in CreateTensorFromBoundingBoxList()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_kernel.h | 266 auto tensor_out = mindspore::MSTensor(impl); in set_out_tensor() local
267 kernel_->set_output(tensor_out, index); in set_out_tensor()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 429 auto tensor_out = std::make_shared<AbstractTensor>(tensor_in->element(), in InferImplAllSwap() local
433 tensor_out->element(), std::make_shared<Shape>(tensor_out_shape, min_shape, max_shape)); in InferImplAllSwap()
|