Home
last modified time | relevance | path

Searched defs:tensor_c (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/common/
Dtensor_util.cc28 auto *tensor_c = static_cast<TensorC *>(malloc(sizeof(TensorC))); in OutputTensor2TensorC() local
220 auto *tensor_c = reinterpret_cast<TensorC *>(malloc(sizeof(TensorC))); in GenerateInTensorC() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_cost.cc391 …int64_t tensor_c = static_cast<int64_t>(node.tensor_parm.tensor_shape.shape_c * node.tensor_parm.t… in GetOptimalStr() local
558 …int64_t tensor_c = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_c * op.arguments[0].ten… in GetOptimalStr() local
642 …int64_t tensor_c = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_c * op.arguments[0].ten… in GetOptimalStr() local
/third_party/mindspore/tests/ut/python/model/
Dtest_mix_precision.py210 def construct(self, tensor_c, **kwargs): argument