HomeSort by relevance Sort by last modified time
    Searched refs:ByteSizeOfPrimitiveType (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_instruction_fusion.cc 25 return ShapeUtil::ByteSizeOfPrimitiveType(shape.element_type()) *
vector_support_library.cc 220 pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_), name());
229 pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_), name());
239 value, pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_));
250 value, pointer, ShapeUtil::ByteSizeOfPrimitiveType(primitive_type_));
ir_emitter.cc 247 int64 byte_size = ShapeUtil::ByteSizeOfPrimitiveType(primitive_type);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util_test.cc 275 EXPECT_EQ(4, ShapeUtil::ByteSizeOfPrimitiveType(F32));
279 EXPECT_EQ(8, ShapeUtil::ByteSizeOfPrimitiveType(F64));
283 EXPECT_EQ(8, ShapeUtil::ByteSizeOfPrimitiveType(C64));
289 EXPECT_EQ(4, ShapeUtil::ByteSizeOfPrimitiveType(F32));
shape_util.h 172 static int64 ByteSizeOfPrimitiveType(PrimitiveType primitive_type);
shape_util.cc 660 /* static */ int64 ShapeUtil::ByteSizeOfPrimitiveType(
744 ByteSizeOfPrimitiveType(shape.element_type());
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_helpers.cc 66 xla::ShapeUtil::ByteSizeOfPrimitiveType(xla_output_type) * 8 - 1;

Completed in 319 milliseconds