HomeSort by relevance Sort by last modified time
    Searched defs:ByteSizeOf (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_rematerialization_test.cc 134 static int64 ByteSizeOf(const Shape& shape) {
135 return ShapeUtil::ByteSizeOf(shape, sizeof(void*));
163 ByteSizeOf,
196 ByteSizeOf,
237 ByteSizeOf,
273 ByteSizeOf,
315 ByteSizeOf,
386 ByteSizeOf,
387 /*memory_limit_bytes=*/4 * ByteSizeOf(vec1024_shape_),
481 ByteSizeOf,
    [all...]
buffer_assignment_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
llvm_util.cc 606 int64 ByteSizeOf(const Shape& shape, const llvm::DataLayout& data_layout) {
608 return ShapeUtil::ByteSizeOf(shape, pointer_size);
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc 176 ByteSizeOf(literal.shape()) >= kMaxInternalConstantSizeInBytes) {
255 int64 IrEmitter::ByteSizeOf(const Shape& shape) const {
256 return llvm_ir::ByteSizeOf(shape, module_->getDataLayout());
265 int64 buffer_size = ByteSizeOf(shape);
290 AttachDereferenceableMetadataForLoad(load, ByteSizeOf(shape));
380 int64 length = ByteSizeOf(shape);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.cc 700 /* static */ int64 ShapeUtil::ByteSizeOf(const Shape& shape,
    [all...]

Completed in 242 milliseconds