Home
last modified time | relevance | path

Searched refs:WritesMlirBuffer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h256 bool WritesMlirBuffer(mlir::Operation* op, mlir::Value operand);
Dir_emission_utils.cc665 if (!WritesMlirBuffer(op, op->getOperand(i)) || aliased) { in PartitionLmhloOperandsAndOutputs()
711 bool WritesMlirBuffer(mlir::Operation* op, mlir::Value operand) { in WritesMlirBuffer() function
Dir_emitter_unnested.cc3525 slice.written = WritesMlirBuffer(op, operand); in BuildKernelThunkForMlir()