Home
last modified time | relevance | path

Searched refs:tf_function (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/cc/experimental/libtf/tests/
Dfunction_test.cc155 Function tf_function; in TEST_P() local
158 Status s = tf_function.RegisterTrace(std::move(trace), signature, signature); in TEST_P()
161 StatusOr<TaggedValue> v = tf_function.Execute(ctx_.get(), args); in TEST_P()
173 Function tf_function; in TEST_P() local
180 tf_function.RegisterTrace(std::move(trace), input_signature, tensor_spec); in TEST_P()
185 StatusOr<TaggedValue> v = tf_function.Execute(ctx_.get(), args); in TEST_P()
196 Function tf_function; in TEST_P() local
202 Status s = tf_function.RegisterTrace(std::move(trace), signature, signature); in TEST_P()
207 StatusOr<TaggedValue> v = tf_function.Execute(ctx_.get(), args); in TEST_P()
219 Function tf_function; in TEST_P() local
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dstrategy_common_test.py253 tf_function=[True, False],
257 def testDenseUpdate(self, strategy, tf_function, update_fn): argument
258 if isinstance(strategy, tpu_strategy.TPUStrategy) and (not tf_function):
277 if tf_function:
296 tf_function=[combinations.tf_function, combinations.no_tf_function],
300 def testDense(self, strategy, tf_function): argument
302 tf_function is combinations.no_tf_function):
305 @tf_function
319 def testSparse(self, strategy, tf_function): argument
320 if tf_function is combinations.no_tf_function:
[all …]
Dvalues_v2_test.py390 tf_function=[combinations.tf_function, combinations.no_tf_function]))
406 def testRead(self, strategy, enable_packed_handle, tf_function): argument
414 @tf_function
419 @tf_function
424 @tf_function
433 def testAssign(self, strategy, enable_packed_handle, tf_function): argument
436 @tf_function
441 @tf_function
455 def testStrategyRun(self, strategy, enable_packed_handle, tf_function): argument
457 tf_function is combinations.no_tf_function):
[all …]
Dcombinations_test.py182 tf_function_1=combinations.tf_function,
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_tf_functions_test.cc83 const TfFunction& tf_function = in TEST() local
85 EXPECT_EQ(tf_function.total_tracing_count(), 4); in TEST()
86 EXPECT_EQ(tf_function.compiler(), MIXED_COMPILER); in TEST()
87 EXPECT_NEAR(tf_function.expensive_call_percent(), 90, kMaxError); in TEST()
89 const auto& metrics = tf_function.metrics(); in TEST()
Dxplane_to_tf_functions.cc62 double ComputeExpensiveCallPercent(const TfFunction& tf_function) { in ComputeExpensiveCallPercent() argument
66 for (const auto& mode_metrics : tf_function.metrics()) { in ComputeExpensiveCallPercent()
/external/tensorflow/tensorflow/cc/experimental/libtf/runtime/
Druntime.cc111 Function tf_function; in ImportModule() local
151 Status status = tf_function.RegisterTrace( in ImportModule()
154 TaggedValue callable = MakeCallable(child_name, tf_function, ctx); in ImportModule()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc205 TF_Function* tf_function = new TF_Function(); in TF_GraphToFunctionWithControlOutputs() local
212 control_output_names_vec, description, &tf_function->fdef); in TF_GraphToFunctionWithControlOutputs()
214 TF_DeleteFunction(tf_function); in TF_GraphToFunctionWithControlOutputs()
219 tf_function->stack_traces[n->name()] = n->GetStackTrace(); in TF_GraphToFunctionWithControlOutputs()
222 return tf_function; in TF_GraphToFunctionWithControlOutputs()
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/
Dcategory_hash_varlen_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
53 @tf_function()
Dbucketized_column_dense_benchmark.py21 from tensorflow.python.eager.def_function import function as tf_function unknown
54 @tf_function()
Dcategory_hash_dense_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
52 @tf_function()
Dembedding_dense_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
54 @tf_function()
Dcategory_vocab_list_dense_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
50 @tf_function()
Dembedding_varlen_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
55 @tf_function()
Dcategory_vocab_list_varlen_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
53 @tf_function()
Dcategory_vocab_list_indicator_dense_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
56 @tf_function()
Dcategory_vocab_list_indicator_varlen_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
59 @tf_function()
Dweighted_embedding_varlen_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
63 @tf_function()
Dcategory_cross_hash_dense_benchmark.py19 from tensorflow.python.eager.def_function import function as tf_function unknown
56 @tf_function()
Dcategory_vocab_file_dense_benchmark.py21 from tensorflow.python.eager.def_function import function as tf_function unknown
68 @tf_function()
Dcategory_vocab_file_varlen_benchmark.py21 from tensorflow.python.eager.def_function import function as tf_function unknown
71 @tf_function()
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dvalues.py33 from tensorflow.python.eager import function as tf_function unknown
338 elif not isinstance(dataset_fn, tf_function.ConcreteFunction):
378 if not isinstance(self._dataset_fn, tf_function.ConcreteFunction):
Dcluster_coordinator.py40 from tensorflow.python.eager import function as tf_function unknown
188 elif isinstance(function, tf_function.ConcreteFunction):
1031 (def_function.Function, tf_function.ConcreteFunction)):
/external/tensorflow/tensorflow/core/profiler/protobuf/
DBUILD131 srcs = ["tf_function.proto"],
Dop_stats.proto9 import "tensorflow/core/profiler/protobuf/tf_function.proto";

12