Home
last modified time | relevance | path

Searched defs:tensor_size (Results 1 – 25 of 30) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Doptimizer_kernel.h73 size_t tensor_size = in InitServerKernelInputOutputSize() local
80 size_t tensor_size = in InitServerKernelInputOutputSize() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc112 …size_t tensor_size = std::accumulate(data_shape.begin(), data_shape.end(), type_size, std::multipl… in AssignValueNodeAddress() local
144 size_t tensor_size = in AssignInputNodeAddress() local
213 …size_t tensor_size = std::accumulate(temp_shape.begin(), temp_shape.end(), type_size, std::multipl… in CreatTensorForOutput() local
229 …size_t tensor_size = std::accumulate(data_shape.begin(), data_shape.end(), type_size, std::multipl… in CreatTensorForOutput() local
336 size_t tensor_size = std::accumulate(data_shape.begin(), data_shape.end(), in BindInputTensorAddressPtr() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Dbatch_norm_grad_cpu_kernel.cc35 size_t tensor_size = shape[C] * SCALE_SHIFT_NUM * type_size; in InitInputOutputSize() local
Dbatch_norm_cpu_kernel.cc35 size_t tensor_size = shape[1] * 2 * type_size; // [2, c] to store scale and bias in InitInputOutputSize() local
Dsoftmax_cross_entropy_with_logits_cpu_kernel.cc38 …size_t tensor_size = std::accumulate(shape.begin(), shape.end(), type_size, std::multiplies<size_t… in InitInputOutputSize() local
Dsparse_softmax_cross_entropy_with_logits_cpu_kernel.cc39 …size_t tensor_size = std::accumulate(shape.begin(), shape.end(), type_size, std::multiplies<size_t… in InitInputOutputSize() local
/third_party/mindspore/mindspore/lite/src/
Dlite_model.cc200 auto tensor_size = this->graph_.all_tensors_.size(); in NodeVerify() local
231 auto tensor_size = this->graph_.all_tensors_.size(); in SubGraphVerify() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/util/
Djson_helper.cc136 size_t JsonHelper::DumpData(const unsigned char *tensor_addr, const size_t &tensor_size, void *addr, in DumpData()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc98 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(item, index); in CreateParameterDeviceAddress() local
132 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(value_node, output_idx); in CreateDeviceAddressForTensorValue() local
161 size_t tensor_size = value.size(); in CreateValueNodeDeviceAddress() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc232 auto tensor_size = runtime_info->output_tensor_size(i); in RunOpMallocPre() local
294 auto tensor_size = AnfAlgo::GetOutputTensorMemSize(input_node, index); in ResetNodeAddress() local
422 auto tensor_size = AnfAlgo::GetOutputTensorMemSize(item, index); in RunOpAssignInputMemory() local
618 auto tensor_size = AnfAlgo::GetOutputTensorMemSize(item, index); in AssignStaticMemoryInput() local
971 size_t tensor_size = LongToSize(tensor->data().nbytes()); in AssignValueNodeTensor() local
1039 size_t tensor_size = value.size(); in AssignStaticMemoryValueNode() local
1407 auto tensor_size = LongToSize(tensor->data().nbytes()); in InitGraphInputTensors() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc313 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(input_node, 0); in PrepareDataForStepMode() local
391 size_t tensor_size = value.size(); in PrepareDataForValueNode() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel.cc35 size_t tensor_size = in InitInputOutputSize() local
45 size_t tensor_size = in InitInputOutputSize() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dbatchmatmul_fusion.cc63 auto tensor_size = fc_weight_param->Size(); in GetRightMatmulInputParamter() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_dynamic_allocator.cc164 bool DynamicMemPoolBestFit::IsDivide(size_t tensor_size, size_t mem_buf_size) const { in IsDivide()
Dmem_swap_manager.cc82 size_t tensor_size = tensor_info.tensor_size_; in InitSwapThreshold() local
262 size_t tensor_size = tensor.tensor_size_; in AddSwapInfo() local
Dmem_reuse_checker.cc74 size_t tensor_size = in CalculOriInput() local
/third_party/mindspore/mindspore/lite/java/native/runtime/
Dms_tensor.cpp276 int tensor_size = static_cast<jint>(data_len / sizeof(float)); in Java_com_mindspore_lite_MSTensor_createTensor() local
Dlite_session.cpp261 auto tensor_size = static_cast<int>(env->GetArrayLength(dims)); in Java_com_mindspore_lite_LiteSession_resize() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dnode_infershape.cc445 auto tensor_size = data_info.data_.size(); in ConvertToLiteTensor() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_cost.cc80 double tensor_size, bool search_forward) { in CostRedisWithAdjacentNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddata_helper.cc177 size_t DataHelper::DumpData(const unsigned char *tensor_addr, const size_t &tensor_size, void *addr, in DumpData()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dcommunication_op_fusion.cc358 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(input_node, 0); in CreateFusedCommunicationOp() local
/third_party/mindspore/mindspore/lite/tools/benchmark_train/
Dnet_train.cc566 int tensor_size = tensor->ElementsNum(); in CheckSum() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.cc1046 auto tensor_size = AnfAlgo::GetOutputTensorMemSize(node, i); in CacheCNodeOutputInfo() local
1071 auto tensor_size = AnfAlgo::GetOutputTensorMemSize(input, index); in CacheCNodeOutputInfo() local
1745 size_t tensor_size = LongToSize(front_tensor->data().nbytes()); in SyncInitialTenosrToDevice() local
/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dgraph_split.py22 def tensor_size(tensor): function

12