Searched refs:ShapeToIrType (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 57 llvm_ir::ShapeToIrType(param_shape, module_)->getPointerTo()); in EmitBasePointersForNestedComputation()
67 llvm_ir::ShapeToIrType(root_shape, module_)->getPointerTo()); in EmitBasePointersForNestedComputation()
|
| D | hlo_to_ir_bindings.cc | 112 llvm_ir::ShapeToIrType(non_io_hlo->shape(), module_); in EmitBasePointersForHlos()
156 llvm::Type* pointee_type = llvm_ir::ShapeToIrType( in GetTypedIrValue()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 110 llvm::Type* element_type = ShapeToIrType(target_shape, module); in EmitGetTupleElement()
|
| D | llvm_util.h | 136 llvm::Type* ShapeToIrType(const Shape& shape, llvm::Module* module);
|
| D | fused_ir_emitter.cc | 96 llvm_ir::ShapeToIrType(literal.shape(), module_)->getPointerTo()); in HandleConstant()
|
| D | ir_array.cc | 414 llvm::Type* new_ir_type = llvm_ir::ShapeToIrType(new_shape, module); in CastToShape()
|
| D | llvm_util.cc | 223 llvm::Type* ShapeToIrType(const Shape& shape, llvm::Module* module) { in ShapeToIrType() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 892 llvm_ir::ShapeToIrType(new_shape, module)->getPointerTo()); in CollapseFirstNDims()
923 llvm_ir::ShapeToIrType(inner_shape, module)->getPointerTo(); in SliceOutInnerArray()
|
| D | ir_emitter.cc | 2820 return llvm_ir::ShapeToIrType(shape, module_); in IrShapeType()
|