Searched refs:MinimumAlignmentForShape (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 152 /*alignment=*/MinimumAlignmentForShape(element_shape), tuple_lvalue, in EmitThreadLocalFunctionEpilogue()
158 /*alignment=*/MinimumAlignmentForShape(element_shape), in EmitThreadLocalFunctionEpilogue()
272 llvm::Align(MinimumAlignmentForShape(literal.shape()))); in EmitGlobalForLiteral()
342 int IrEmitter::MinimumAlignmentForShape(const Shape& shape) { in MinimumAlignmentForShape() function in xla::cpu::IrEmitter
356 int alignment = MinimumAlignmentForShape(shape); in AttachAlignmentMetadataForLoad()
391 shape, get_tuple_element->tuple_index(), MinimumAlignmentForShape(shape), in HandleGetTupleElement()
550 tuple_element_shape, i, MinimumAlignmentForShape(tuple_element_shape), in HandleOutfeed()
3267 &b_, MinimumAlignmentForShape(target_shape)); in EmitThreadLocalBufferPointer()
|
| H A D | ir_emitter.h | 626 int MinimumAlignmentForShape(const Shape& shape);
|