Searched refs:use_mlir (Results 1 – 5 of 5) sorted by relevance
65 def testAdd(self, use_mlir): argument66 if use_mlir:86 def testAddGrad(self, use_mlir): argument87 if use_mlir:114 def testRelu(self, use_mlir): argument115 if use_mlir:140 def testReluGrad(self, use_mlir): argument141 if use_mlir:170 def testNeg(self, use_mlir): argument171 if use_mlir:[all …]
343 bool use_mlir = computation.index() == 0; in CreateTpuCompilationRequest() local344 compilation_request.set_use_mlir(use_mlir); in CreateTpuCompilationRequest()345 if (use_mlir) { in CreateTpuCompilationRequest()
33 bool use_mlir = 2; field
797 def test_enqueue_cpu_tensor_with_outside_compilation(self, use_mlir): argument798 if use_mlir:865 def test_enqueue_with_outside_compilation(self, use_mlir): argument866 if use_mlir:903 def test_enqueue_with_outside_compilation_in_control_flow(self, use_mlir): argument904 if use_mlir:
288 bool use_mlir = config && in CompileSingleOp() local294 if (!use_mlir) { in CompileSingleOp()