Searched refs:tf_function (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_common_test.py | 121 tf_function=[combinations.tf_function, combinations.no_tf_function],
125 def testBasic(self, strategy, tf_function): argument
127 tf_function is combinations.no_tf_function):
130 @tf_function
143 def testNestedInput(self, strategy, tf_function): argument
148 tf_function is combinations.no_tf_function):
151 @tf_function
|
| D | combinations_test.py | 182 tf_function_1=combinations.tf_function,
|
| D | combinations.py | 316 tf_function = combinations_lib.NamedObject("TfFunction", def_function.function) variable
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_tf_functions_test.cc | 83 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()
|
| D | xplane_to_tf_functions.cc | 62 double ComputeExpensiveCallPercent(const TfFunction& tf_function) { in ComputeExpensiveCallPercent() argument
66 for (const auto& mode_metrics : tf_function.metrics()) { in ComputeExpensiveCallPercent()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 205 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/distribute/coordinator/ |
| D | cluster_coordinator.py | 43 from tensorflow.python.eager import function as tf_function unknown
365 elif isinstance(function, tf_function.ConcreteFunction):
1104 (def_function.Function, tf_function.ConcreteFunction)):
1322 elif not isinstance(dataset_fn, tf_function.ConcreteFunction):
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | BUILD | 131 srcs = ["tf_function.proto"],
|
| D | op_stats.proto | 9 import "tensorflow/core/profiler/protobuf/tf_function.proto";
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 91 tf_function = def_function.function(f)
94 functools.partial(_jvp, tf_function, primal), tangents)
|
| D | function_test.py | 49 from tensorflow.python.framework import function as tf_function unknown
1657 @tf_function.Defun(dtypes.int32, dtypes.int32)
|