| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runtime_info.cc | 51 size_t tensor_size = type_size * SizeOf(shape); in OpRuntimeInfoGetOutputTensorMemSize() local
69 …auto tensor_size = OpRuntimeInfoGetOutputTensorMemSize(node, i, output_type, output_format, device… in CacheForExecutionOrder() local
117 …auto tensor_size = OpRuntimeInfoGetOutputTensorMemSize(input, index, type_id, format, device_shape… in CacheForGraphInputs() local
166 void OpRuntimeInfo::SetOutputTensorSize(size_t index, size_t tensor_size) { in SetOutputTensorSize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_kernel_runtime.cc | 113 …size_t tensor_size = std::accumulate(data_shape.begin(), data_shape.end(), type_size, std::multipl… in AssignValueNodeAddress() local
146 size_t tensor_size = type_size * SizeOf(fmt_shape); in AssignInputNodeAddress() local
213 …size_t tensor_size = std::accumulate(temp_shape.begin(), temp_shape.end(), type_size, std::multipl… in CreateTensorForOutput() local
230 …size_t tensor_size = std::accumulate(data_shape.begin(), data_shape.end(), type_size, std::multipl… in CreateTensorForOutput() local
340 size_t tensor_size = std::accumulate(data_shape.begin(), data_shape.end(), in BindInputTensorAddressPtr() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | device_address_utils.cc | 55 size_t tensor_size = value.size() + 1; in CreateDeviceAddressForScalarAndString() local
245 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(item, index); in CreateParameterDeviceAddress() local
311 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(value_node, output_idx); in CreateDeviceAddressForTensorValue() local
587 size_t tensor_size = type_size * SizeOf(device_shape); in GetTensorDeviceSize() local
872 auto tensor_size = LongToSize(tensor->data().nbytes()); in CreateInputTensorAddress() local
925 auto tensor_size = LongToSize(tensor->data().nbytes()); in MallocForInput() local
1043 const auto &tensor_size = LongToSize(tensor->data().nbytes()); in CreateInputAddress() local
1162 auto tensor_size = LongToSize(tensor->data().nbytes()); in CreateOutputTensorAddress() local
1204 auto tensor_size = GetTypeByte(TypeIdToType(tensor->data_type())) * SizeOf(real_shape); in CreateDeviceAddress() local
|
| D | tensor_array.cc | 82 auto tensor_size = tensors_[i]->size; in Write() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | convert_to_dynamic_cpu_kernel.cc | 44 size_t tensor_size = 0; in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | bessel_i1_cpu_kernel.cc | 73 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_k0_cpu_kernel.cc | 73 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_k1e_cpu_kernel.cc | 74 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_y1_cpu_kernel.cc | 73 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_j0_cpu_kernel.cc | 72 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_i0e_cpu_kernel.cc | 74 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_y0_cpu_kernel.cc | 73 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_i0_cpu_kernel.cc | 73 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_j1_cpu_kernel.cc | 73 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_i1e_cpu_kernel.cc | 74 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_k1_cpu_kernel.cc | 73 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| D | bessel_k0e_cpu_kernel.cc | 74 size_t tensor_size = inputs[0]->size() / sizeof(T); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | masked_select_grad.cc | 104 uint64_t tensor_size = 1; in MaskedSelectGradCompute() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/common/ |
| D | device_common_test.h | 143 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(input_node, index); in CreateKernel() local
152 size_t tensor_size = AnfAlgo::GetOutputTensorMemSize(node, output_index); in CreateKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | triu_indices_gpu_kernel.cc | 46 size_t tensor_size = 0; in Resize() local
|
| D | tril_indices_gpu_kernel.cc | 46 size_t tensor_size = 0; in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | priority_replay_buffer.cc | 115 …size_t tensor_size = std::accumulate(shapes[i].begin(), shapes[i].end(), type_size, std::multiplie… in Init() local
205 …size_t tensor_size = std::accumulate(shapes[i].begin(), shapes[i].end(), type_size, std::multiplie… in Init() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/ |
| D | tensor_array_stack_kernel.cc | 71 size_t tensor_size = tensors_->GetValidSize(); in UpdateOutputShapeAndSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rl/ |
| D | tensor_array_stack_kernel.cc | 68 size_t tensor_size = tensors_->GetValidSize(); in UpdateOutputShapeAndSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/acl_kernel/ |
| D | custom_op_kernel_mod.cc | 86 …size_t tensor_size = std::accumulate(shape.begin(), shape.end(), type_size, std::multiplies<size_t… in Resize() local
|