Home
last modified time | relevance | path

Searched refs:batch_function (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dbatch_ops_test.py220 @batch_ops.batch_function(1, 10, 100000)
269 @batch_ops.batch_function(1, 10, 100000)
294 @batch_ops.batch_function(1, 10, 100000)
320 @batch_ops.batch_function(1, 2, 1)
343 @batch_ops.batch_function(1, 10, 100000)
378 result = gen_batch_ops.batch_function(
411 result = gen_batch_ops.batch_function(
445 result = gen_batch_ops.batch_function(
472 result = gen_batch_ops.batch_function(
521 @batch_ops.batch_function(1, 10, 100000)
H A Dbatch_ops.py29 def batch_function(num_batch_threads, function
104 outputs = gen_batch_ops.batch_function(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
H A Dbatch_function_fallback_resource_variable_as_captured_tensor.mlir10 … // CHECK: tfrt_fallback_async.batch_function([[ARG:%.*]]) @batched_func ([[BATCHED_FUNC_ARG:%.*]])
H A Dbatch_function_fallback.mlir14 // CHECK: tfrt_fallback_async.batch_function(%arg0) @batched_function ({{%.*}}, [[var]])
H A Dlower_saved_model.mlir142 func @batch_function(
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dtraining_generator_v1.py150 batch_function = _make_execution_function(
256 batch_outs = batch_function(*batch_data)
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_BatchFunction.pbtxt110 b = gen_batch_ops.batch_function(
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
H A Dtfrt_fallback_async.td442 def BatchFunctionOp : FallbackAsync_Op<"batch_function", [CoreRT_TypedAttributeTrait]> {
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
H A Dtfrt_fallback_async.td441 def BatchFunctionOp : FallbackAsync_Op<"batch_function", [CoreRT_TypedAttributeTrait]> {
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dpy_func_test.py470 @batch_ops.batch_function(1, 10, 100000)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td1054 b = gen_batch_ops.batch_function(