Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.h72 xla::StatusOr<Operation*> EmitDnnBatchNorm(
Dmhlo_to_lhlo_with_xla.cc632 return EmitDnnBatchNorm(custom_call_instr); in EmitCustomCallOp()
845 StatusOr<Operation*> LhloDialectEmitter::EmitDnnBatchNorm( in EmitDnnBatchNorm() function in mlir::LhloDialectEmitter