Searched refs:ShapeToIrType (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 65 llvm_ir::ShapeToIrType(param_shape, module_)->getPointerTo()); in CodegenNestedComputation()
75 llvm_ir::ShapeToIrType(root_shape, module_)->getPointerTo()); in CodegenNestedComputation()
139 llvm::Type* tuple_type = llvm_ir::ShapeToIrType(return_shape, module_); in CodegenNestedComputation()
|
| D | hlo_to_ir_bindings.cc | 91 llvm_ir::ShapeToIrType(non_io_hlo->shape(), module_); in EmitBasePointersForHlos()
125 llvm_ir::ShapeToIrType(shape, b->GetInsertBlock()->getModule()); in CastToTypedValue()
|
| D | ir_emitter.cc | 640 llvm_ir::ShapeToIrType(return_shape, module_), "return_buffer", &b_); in ComputeNestedElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 103 llvm_ir::ShapeToIrType(literal.shape(), module_)->getPointerTo()); in HandleConstant()
|
| D | tuple_ops.cc | 126 llvm::Type* element_type = ShapeToIrType(target_shape, module); in EmitGetTupleElement()
|
| D | llvm_util.h | 139 llvm::Type* ShapeToIrType(const Shape& shape, llvm::Module* module);
|
| D | ir_array.cc | 524 llvm::Type* new_ir_type = llvm_ir::ShapeToIrType(new_shape, module); in CastToShape()
|
| D | llvm_util.cc | 231 llvm::Type* ShapeToIrType(const Shape& shape, llvm::Module* module) { in ShapeToIrType() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1088 llvm_ir::ShapeToIrType(new_shape, module)->getPointerTo()); in CollapseFirstNDims()
1119 llvm_ir::ShapeToIrType(inner_shape, module)->getPointerTo(); in SliceOutInnerArray()
|
| D | ir_emitter.cc | 131 llvm::Type* tuple_type = llvm_ir::ShapeToIrType(return_shape, module_); in EmitThreadLocalFunctionEpilogue()
3010 return llvm_ir::ShapeToIrType(shape, module_); in IrShapeType()
3255 llvm_ir::ShapeToIrType(return_shape, module_); in EmitThreadLocalCall()
|