Home
last modified time | relevance | path

Searched defs:element_size_in_bytes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuWinogradConv2dKernel.cpp46 int element_size_in_bytes = input_nhwc->info()->element_size(); in run_op() local
85 const int element_size_in_bytes = dst_nhwc->info()->element_size(); in run_op() local
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_numpy.cpp171 auto element_size_in_bytes = prepared_tensor.element_size(); in tensor_to_numpy() local
238 auto element_size_in_bytes = PyArray_ITEMSIZE(array); in tensor_from_numpy() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp1275 auto element_size_in_bytes = op.tensor_base().element_size(); in compute_strides() local
1411 auto element_size_in_bytes = op.tensor_base().element_size(); in fast_set_up() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuWinogradConv2d.cpp384 const int element_size_in_bytes = permuted_weights.get()->info()->element_size(); in prepare() local