Searched refs:batch_function (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops_test.py | 221 @batch_ops.batch_function(1, 10, 100000)
250 @batch_ops.batch_function(1, 10, 100000)
275 @batch_ops.batch_function(1, 10, 100000)
301 @batch_ops.batch_function(1, 2, 1)
324 @batch_ops.batch_function(1, 10, 100000)
359 result = gen_batch_ops.batch_function(
392 result = gen_batch_ops.batch_function(
426 result = gen_batch_ops.batch_function(
453 result = gen_batch_ops.batch_function(
502 @batch_ops.batch_function(1, 10, 100000)
|
| D | batch_ops.py | 33 def batch_function(num_batch_threads, function
108 outputs = gen_batch_ops.batch_function(
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | batch_function_fallback_resource_variable_as_captured_tensor.mlir | 10 … // CHECK: tfrt_fallback_async.batch_function([[ARG:%.*]]) @batched_func ([[BATCHED_FUNC_ARG:%.*]])
|
| D | batch_function_fallback.mlir | 14 // CHECK: tfrt_fallback_async.batch_function(%arg0) @batched_function ({{%.*}}, [[var]])
|
| D | lower_saved_model.mlir | 142 func @batch_function(
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_generator_v1.py | 150 batch_function = _make_execution_function(
256 batch_outs = batch_function(*batch_data)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchFunction.pbtxt | 110 b = gen_batch_ops.batch_function(
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_async.td | 442 def BatchFunctionOp : FallbackAsync_Op<"batch_function", [CoreRT_TypedAttributeTrait]> {
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | py_func_test.py | 472 @batch_ops.batch_function(1, 10, 100000)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 978 b = gen_batch_ops.batch_function(
|