Home
last modified time | relevance | path

Searched defs:ByteSizeOf (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_rematerialization_test_utils.h139 static int64_t ByteSizeOf(const Shape& shape) { in ByteSizeOf() function
H A Dbuffer_assignment_test.cc2127 static int64_t ByteSizeOf(const BufferValue& buffer) { in ByteSizeOf() function in xla::__anond1a23f2e0111::WhileBufferAssignmentTest
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.h357 int64_t ByteSizeOf(const Shape& shape) const { in ByteSizeOf() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
H A Dobject.h121 inline size_t ByteSizeOf(const Object& object) { in ByteSizeOf() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dllvm_util.cc502 int64_t ByteSizeOf(const Shape& shape, const llvm::DataLayout& data_layout) { in ByteSizeOf() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.cc823 /* static */ int64_t ShapeUtil::ByteSizeOf(const Shape& shape, in ByteSizeOf() function in xla::ShapeUtil
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc337 int64_t IrEmitter::ByteSizeOf(const Shape& shape) const { in ByteSizeOf() function in xla::cpu::IrEmitter