Home
last modified time | relevance | path

Searched refs:IrShapeType (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc171 IrShapeType(bitcast->shape())->getPointerTo(), IrName(bitcast)); in HandleBitcast()
188 result_global, IrShapeType(literal.shape())->getPointerTo()); in EmitGlobalForLiteral()
2819 llvm::Type* IrEmitter::IrShapeType(const Shape& shape) { in IrShapeType() function in xla::cpu::IrEmitter
2883 IrShapeType(shape), absl::StrCat("thread_local", slice.ToString()), in EmitThreadLocalBufferPointer()
2891 return BitCast(tempbuf_address, IrShapeType(target_shape)->getPointerTo()); in EmitThreadLocalBufferPointer()
2916 IrShapeType(target_shape)->getPointerTo()); in EmitGlobalBufferPointer()
2926 IrShapeType(target_shape)->getPointerTo()); in EmitBufferPointer()
Dir_emitter.h247 llvm::Type* IrShapeType(const Shape& shape);