Home
last modified time | relevance | path

Searched defs:jitrt (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/
H A Dtf_metadata_ops_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_strided_slice_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_mean_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_broadcast_to_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_select_test.py28 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_const_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_controlflow_test.py28 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_softmax_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dmultiple_results_test.py28 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_cast_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_reshape_test.py28 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_log1p_test.py45 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_function_test.py31 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_acos_test.py45 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_logical_ops_test.py38 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_matmul_test.py35 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_pack_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_binary_bcast_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_math_ops_test.py27 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_transpose_test.py28 jitrt = tf_jitrt.TfJitRtExecutor() variable
H A Dtf_reduction_test.py22 jitrt = tf_jitrt.TfJitRtExecutor() variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
H A Dtf_jitrt_test.py31 jitrt = tf_jitrt.TfJitRtExecutor() variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/regression_tests/
H A Dcompile_and_run_test.py45 jitrt = tf_jitrt.TfJitRtExecutor() variable