Home
last modified time | relevance | path

Searched defs:input_data_size (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/debug/
Dassert_gpu_kernel.cc44 auto input_data_size = inputs_size - 1; in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Daddcmul_helper.h95 int64_t input_data_size = static_cast<int64_t>(input_data_shape_.size()); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dmemcpy_gpu_kernel.cc48 size_t input_data_size = abstract::TypeIdSize(inputs[kIndex0]->dtype_id()); in Resize() local
/third_party/mesa3d/src/gallium/targets/teflon/
Dtflite-stub.c52 size_t input_data_size) in TfLiteTensorCopyFromBuffer()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Dallocator_utils.cc42 … DataType type, const void *data_ptr, int64_t input_data_size, in UpdateOutputDataTensor()