Searched refs:mlir_context_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 52 mlir_context_(mlir_context), in IrEmitterContext()
69 mlir::MLIRContext* mlir_context() { return mlir_context_; } in mlir_context()
95 mlir::MLIRContext* mlir_context_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 229 mlir::MLIRContext* mlir_context_; member in xla::cpu::__anonf3b678a30111::DotOpEmitter
250 mlir_context_(mlir_context), in DotOpEmitter()
271 mlir_context_, b_, dot_info_.result_shape, operand_shapes, target_ptr, in EmitLinalgMatmul()
|
| D | ir_emitter.h | 456 mlir::MLIRContext* mlir_context_; variable
|
| D | ir_emitter.cc | 102 mlir_context_(mlir_context), in IrEmitter()
866 GetExecutableRunOptionsArgument(), &b_, mlir_context_, in HandleDot()
2109 GetExecutableRunOptionsArgument(), &b_, mlir_context_, in HandleFusion()
|