Home
last modified time | relevance | path

Searched refs:tf_jitrt (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/
DBUILD15 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
27 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
43 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
59 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
74 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
89 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
104 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
116 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
131 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
143 "//tensorflow/compiler/mlir/tfrt/jit/python_binding:tf_jitrt",
[all …]
Dtf_log1p_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
45 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_select_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
28 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_acos_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
45 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_logical_ops_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
38 jitrt = tf_jitrt.TfJitRtExecutor()
Dmultiple_results_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
28 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_controlflow_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
28 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_function_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
26 tf_jitrt.Specialization.ALWAYS,
31 jitrt = tf_jitrt.TfJitRtExecutor()
91 tf_jitrt.Specialization.ALWAYS, vectorize)
122 tf_jitrt.Specialization.ALWAYS, vectorize)
157 tf_jitrt.Specialization.ALWAYS, vectorize)
Dtf_reshape_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
28 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_transpose_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
23 tf_jitrt.Specialization.ENABLED,
24 tf_jitrt.Specialization.DISABLED,
25 tf_jitrt.Specialization.ALWAYS,
28 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_binary_bcast_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
25 tf_jitrt.Specialization.ENABLED,
26 tf_jitrt.Specialization.DISABLED,
27 tf_jitrt.Specialization.ALWAYS,
Dtf_strided_slice_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_metadata_ops_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_const_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_mean_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_broadcast_to_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_pack_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_softmax_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
22 jitrt = tf_jitrt.TfJitRtExecutor()
Dtf_matmul_test.py19 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
35 jitrt = tf_jitrt.TfJitRtExecutor()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
Dtf_to_corert_pipeline_cpurt.mlir19 // CHECK: %[[COMPILED:.*]] = tf_jitrt.fallback.compile @kernel::@compute
28 // CHECK: tf_jitrt.fallback.execute @kernel::@compute
56 // CHECK: tf_jitrt.fallback.compile @kernel::@compute
57 // CHECK-NOT: tf_jitrt.fallback.compile
64 // CHECK: tf_jitrt.fallback.execute @kernel::@compute
66 // CHECK: tf_jitrt.fallback.execute @kernel::@compute
98 // CHECK: tf_jitrt.fallback.compile @kernel::@compute
105 // CHECK: tf_jitrt.fallback.execute @kernel::@compute
138 // CHECK: tf_jitrt.fallback.compile @kernel::@compute
145 // CHECK: tf_jitrt.fallback.execute @kernel::@compute
[all …]
Dauto-fusion.mlir11 // CHECK: %[[RES:.*]] = tf_jitrt.fallback.execute @kernel::@compute(%[[ARG]])
33 // CHECK: %[[RES:.*]] = tf_jitrt.fallback.execute @kernel::@compute
55 // CHECK: tf_jitrt.fallback.execute @kernel::@compute
59 // CHECK: tf_jitrt.fallback.execute @kernel_0::@compute
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
DBUILD15 name = "tf_jitrt",
17 srcs = ["tf_jitrt.py"],
31 ":tf_jitrt",
81 "//tensorflow/compiler/mlir/tfrt:tf_jitrt",
Dtf_jitrt_test.py20 from tensorflow.compiler.mlir.tfrt.jit.python_binding import tf_jitrt
31 jitrt = tf_jitrt.TfJitRtExecutor()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/opdefs/
Dtf_jitrt_ops.td33 let name = "tf_jitrt";
42 let cppNamespace = "::mlir::tf_jitrt";
59 `tf_jitrt.fallback.compile` schedules the compilation of a Tensorflow
90 `tf_jitrt.fallback.execute` compiles a Tensorflow program defined by the
112 %result = tf_jitrt.fallback.execute @kernel::@main (%operand)
137 Same as `tf_jitrt.fallback.execute`, except that this kernel allows to pass
175 `tf_jitrt.test.wait_for_compilation` waits for the completion of all
199 `tf_jitrt.test.reset_compilation_thread_pool` resets the compilation thread
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/
Dtf_jitrt_stream_analysis.mlir80 %res = tf_jitrt.fallback.execute @rsqrt_m::@compute (%arg0)
94 %res = tf_jitrt.fallback.execute @add_m::@compute (%arg0, %arg0)
108 %res = tf_jitrt.fallback.execute @fusion_m::@compute (%arg0)
124 %res = tf_jitrt.fallback.execute @dyn_m::@compute (%arg0)
141 %res = tf_jitrt.fallback.execute @dyn_override_m::@compute (%arg0)

12