Searched refs:element_size_in_bits (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout.cc | 58 layout.set_element_size_in_bits(proto.element_size_in_bits()); in CreateFromProto()
73 proto.set_element_size_in_bits(element_size_in_bits()); in ToProto()
84 if (element_size_in_bits() != 0) { in ToString()
85 absl::StrAppend(&colon_string, "E(", element_size_in_bits(), ")"); in ToString()
109 lhs.element_size_in_bits() != rhs.element_size_in_bits()) { in operator ()()
|
| D | layout.h | 94 int64 element_size_in_bits = 0, int64 memory_space = 0)
98 element_size_in_bits_(element_size_in_bits), in format_()
206 int64 element_size_in_bits() const { return element_size_in_bits_; } in element_size_in_bits() function
|
| D | layout_util.cc | 60 int64 element_size_in_bits, int64 memory_space) { in MakeLayout() argument
76 layout.set_element_size_in_bits(element_size_in_bits); in MakeLayout()
456 hash_value = Hash64Combine(hash_value, layout.element_size_in_bits()); in Hash()
|
| D | layout_util.h | 41 int64 element_size_in_bits = 0,
|
| D | shape_util.cc | 123 int64 element_size_in_bits, int64 memory_space) { in MakeShapeWithLayoutInternal() argument
134 if (element_size_in_bits == in MakeShapeWithLayoutInternal()
137 element_size_in_bits = 0; in MakeShapeWithLayoutInternal()
140 minor_to_major, tiles, element_size_in_bits, memory_space); in MakeShapeWithLayoutInternal()
312 int64 element_size_in_bits, int64 memory_space) { in MakeShapeWithLayout() argument
315 tiles, element_size_in_bits, memory_space); in MakeShapeWithLayout()
340 shape.layout().element_size_in_bits()); in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
|
| D | shape_util.h | 465 int64 element_size_in_bits = 0,
|
| D | xla_data.proto | 174 int64 element_size_in_bits = 7; field
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.cc | 273 c_layout->element_size_in_bits = layout.element_size_in_bits(); in ToC()
288 return xla::Layout(minor_to_major, tiles, c_layout->element_size_in_bits, in FromC()
|
| D | c_api_decl.h | 214 int64_t element_size_in_bits; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.cc | 4502 tensorflow::int64 element_size_in_bits = 0; in ParseLayout() local
4538 ParseLayoutIntAttribute(&element_size_in_bits, "element size in bits"); in ParseLayout()
4558 element_size_in_bits, memory_space); in ParseLayout()
|