Home
last modified time | relevance | path

Searched refs:tensor_base (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/ut/python/debugger/gpu_tests/
Dtest_sync_read_tensors_base_stat.py121 for x, (tensor_info_item, tensor_base, tensor_stat) in enumerate(zip(tensor_info,
134 assert tensor_base.data_size == base_json['size_in_bytes']
135 assert tensor_base.dtype == base_json['debugger_dtype']
136 assert tensor_base.shape == base_json['shape']
154 for x, (tensor_info_item, tensor_base, tensor_stat) in enumerate(zip(tensor_info,
169 'size_in_bytes': tensor_base.data_size,
170 'debugger_dtype': tensor_base.dtype,
171 'shape': tensor_base.shape
/third_party/mindspore/mindspore/core/abstract/
Dparam_validator.cc113 ShapePtr CheckShapeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const Abstract… in CheckShapeSame() argument
114 MS_EXCEPTION_IF_NULL(tensor_base); in CheckShapeSame()
115 ShapePtr shape_base = tensor_base->shape(); in CheckShapeSame()
127 TypePtr CheckDtypeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const AbstractT… in CheckDtypeSame() argument
128 MS_EXCEPTION_IF_NULL(tensor_base); in CheckDtypeSame()
129 auto base_elem = tensor_base->element(); in CheckDtypeSame()
Dparam_validator.h44 ShapePtr CheckShapeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const Abstract…
46 TypePtr CheckDtypeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const AbstractT…
Dprim_arrays.cc103 AbstractTensorPtr tensor_base = CheckArg<AbstractTensor>(op_name, arg->elements(), 0); in InferImplStack() local
104 auto shape = tensor_base->shape(); in InferImplStack()
116 (void)CheckDtypeSame(op_name, tensor_base, tensor); in InferImplStack()
117 (void)CheckShapeSame(op_name, tensor_base, tensor); in InferImplStack()
119 auto element = tensor_base->element(); in InferImplStack()
124 AbstractTensorPtr ret = dyn_cast<AbstractTensor>(tensor_base->Broaden()); in InferImplStack()
935 AbstractTensorPtr tensor_base = nullptr; in InferImplConcat() local
942 tensor_base = CheckArg<AbstractTensor>(op_name, arg->elements(), 0); in InferImplConcat()
945 tensor_base = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplConcat()
948 MS_EXCEPTION_IF_NULL(tensor_base); in InferImplConcat()
[all …]
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Dgrpc_client.h56 EventReply SendTensorBase(const std::list<TensorBase> &tensor_base);
Dgrpc_client.cc205 for (const auto &tensor_base : tensor_base_list) { in SendTensorBase() local
206 if (!writer->Write(tensor_base)) { in SendTensorBase()
Ddebug_grpc.proto155 TensorBase tensor_base = 1; field
Ddebugger.cc941 TensorBase *tensor_base = tensor_summary_item.mutable_tensor_base(); in AddTensorStatInfo() local
942 tensor_base->set_data_type(tensor_stat.dtype); in AddTensorStatInfo()
943 tensor_base->set_data_size((int64_t)tensor_stat.data_size); in AddTensorStatInfo()
945 tensor_base->add_shape(elem); in AddTensorStatInfo()