Searched defs:use_mlir (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 66 def testAdd(self, use_mlir): argument
87 def testAddGrad(self, use_mlir): argument
115 def testRelu(self, use_mlir): argument
141 def testReluGrad(self, use_mlir): argument
171 def testNeg(self, use_mlir): argument
191 def testNegGrad(self, use_mlir): argument
215 def testSub(self, use_mlir): argument
236 def testSubGrad(self, use_mlir): argument
264 def testMul(self, use_mlir): argument
285 def testMulGrad(self, use_mlir): argument
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile.proto | 33 bool use_mlir = 2; field
|
| D | tpu_compile_op_support.cc | 348 bool use_mlir = computation.index() == 0; in CreateTpuCompilationRequest() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_test.py | 810 def test_enqueue_cpu_tensor_with_outside_compilation(self, use_mlir): argument
879 def test_enqueue_with_outside_compilation(self, use_mlir): argument
918 def test_enqueue_with_outside_compilation_in_control_flow(self, use_mlir): argument
|