Home
last modified time | relevance | path

Searched defs:element_size_bytes (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dkernel_ops_pattern.cc204 auto element_size_bytes = xla::GetElementTypeBytes(type.getElementType()); in GetAllocations() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh85 int element_size_bytes; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorAdvancedIndexing.cpp555 int64_t element_size_bytes = src.element_size(); in AdvancedIndex() local