Home
last modified time | relevance | path

Searched refs:get_concrete_function (Results 1 – 25 of 128) sorted by relevance

123456

/external/tensorflow/tensorflow/python/eager/
Dfunction_argument_naming_test.py44 fn_op = fn.get_concrete_function(
68 fn_op = fn.get_concrete_function(
87 fn_op = fn.get_concrete_function(
99 fn_op2 = fn.get_concrete_function(
113 fn_op3 = fn.get_concrete_function(
136 class_op = HasMethod.method.get_concrete_function(
146 method_op = has_method.method.get_concrete_function(
158 method_op = has_method.method.get_concrete_function(
181 method_op = has_method.method.get_concrete_function()
188 method_op2 = has_method.method.get_concrete_function()
[all …]
Dremote_cluster_test.py230 worker_fn.get_concrete_function(x1)
255 worker_fn.get_concrete_function(x1)
283 worker_fn.get_concrete_function(x1)
311 worker_fn.get_concrete_function(x1)
343 concrete_fn = worker_fn.get_concrete_function(x1)
360 worker_fn.get_concrete_function(x1)
434 worker_fn.get_concrete_function(x1)
474 worker_fn.get_concrete_function(x1)
494 worker_fn.get_concrete_function(x1)
527 worker_fn.get_concrete_function(x1)
[all …]
Dfunction_test.py258 arg_attrs = read_var.get_concrete_function().function_def.arg_attr
372 v = v.get_concrete_function(
598 cf = f.get_concrete_function()
754 self.f = f.get_concrete_function()
759 weak_g_graph = weakref.ref(model.g.get_concrete_function().graph)
789 sq_op = sq.get_concrete_function(t)
804 cf = sq.get_concrete_function()
820 f1, f2 = p.map(add_100.get_concrete_function, [args] * 2)
832 sq_op = sq.get_concrete_function(
852 sq_op_autonamed = sq.get_concrete_function(
[all …]
Ddef_function_test.py416 concrete = compute.get_concrete_function()
418 concrete = compute.get_concrete_function(
603 conc = f.get_concrete_function(
609 conc = f.get_concrete_function(tensor_spec.TensorSpec(None, dtypes.float32))
618 conc = g.get_concrete_function(
740 concrete = non_unique_arg_names.get_concrete_function(
791 add_var.get_concrete_function(constant_op.constant(2.))
833 func_a = func.get_concrete_function(
835 func_b = func.get_concrete_function(
846 func_a = func.get_concrete_function(constant_op.constant(2.))
[all …]
Dlift_to_graph_test.py39 concrete_fn = fn.get_concrete_function()
65 concrete_fn = fn.get_concrete_function()
/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops_test.py57 then_cf = Then.get_concrete_function()
58 else_cf = Else.get_concrete_function()
74 then_cf = Then.get_concrete_function()
75 else_cf = Else.get_concrete_function()
Dembedding_ops_test.py47 concrete_call = _call.get_concrete_function()
92 concrete_call = _call.get_concrete_function()
Darray_ops_test.py82 conc = g.get_concrete_function(tensor_spec.TensorSpec([10, None]))
91 conc = g.get_concrete_function(tensor_spec.TensorSpec([10, None]))
/external/tensorflow/tensorflow/core/function/
Druntime_client_test.py114 cf = f.get_concrete_function()
130 cf = f.get_concrete_function()
151 cf = f.get_concrete_function()
171 cf = f.get_concrete_function()
192 cf = f.get_concrete_function(one, one)
/external/tensorflow/tensorflow/lite/python/
Danalyzer_test.py84 [func.get_concrete_function()], func)
102 [func.get_concrete_function()], func)
139 [func.get_concrete_function()], func)
168 root.f1 = add.get_concrete_function()
169 root.f2 = sub.get_concrete_function()
200 [func.get_concrete_function()], func)
218 [func.get_concrete_function()], func)
Dlite_v2_test.py88 concrete_func = root.f.get_concrete_function(input_data)
107 concrete_func = root.f.get_concrete_function(input_data)
124 concrete_func = root.f.get_concrete_function(input_data)
146 concrete_func = root.__call__.get_concrete_function(
176 to_save = root.f.get_concrete_function()
193 concrete_func = root.add.get_concrete_function(input_data)
210 add_func = root.add.get_concrete_function(input_data)
211 sub_func = root.sub.get_concrete_function(input_data)
282 to_save = root.f.get_concrete_function()
395 [func.get_concrete_function()])
[all …]
Dlite_flex_test.py138 concrete_func = root.f.get_concrete_function(input_data)
185 [model.get_concrete_function()])
305 to_save = m.eval.get_concrete_function()
340 concrete_func = root.add_func.get_concrete_function(input_data)
372 concrete_func = root.l2_loss_func.get_concrete_function(input_data)
416 concrete_func = root.conv_func.get_concrete_function(
/external/tensorflow/tensorflow/python/compiler/mlir/
Dmlir_test.py58 tf_graph_def = add_func.get_concrete_function().graph.as_graph_def()
125 concrete_function = sqr.get_concrete_function(
142 concrete_function = caller.get_concrete_function(
155 concrete_function = logging.get_concrete_function()
/external/tensorflow/tensorflow/python/distribute/
Ddistributed_table_test.py346 captures = model.use_table.get_concrete_function().captured_inputs
360 func_captures = model.use_table.get_concrete_function(
365 deferred_captures = model.use_table.get_concrete_function(
374 func_captures = distributed_model.use_table.get_concrete_function(
381 deferred_captures = distributed_model.use_table.get_concrete_function(
405 loaded_without_strategy.use_table.get_concrete_function().graph
408 loaded_without_strategy.use_table.get_concrete_function().graph
437 loaded.use_table.get_concrete_function().graph.external_captures)
439 loaded.use_table.get_concrete_function().graph
450 t for t in loaded.use_table.get_concrete_function().captured_inputs
/external/tensorflow/tensorflow/python/saved_model/
Dtracing_utils.py48 concrete_save = save_fn.get_concrete_function()
61 concrete_restore = restore_fn.get_concrete_function(
Dtracing_utils_test.py53 .get_concrete_function())
56 .get_concrete_function(restored_tensor_spec))
Dload_test.py198 exported_graph = exported.f.get_concrete_function().graph
365 root.f.get_concrete_function(
542 root.f = def_function.function(func).get_concrete_function(input1)
590 root.f = def_function.function(func).get_concrete_function(
1012 func.get_concrete_function(
1014 func.get_concrete_function(tensor_spec.TensorSpec([None], dtypes.float32))
1021 concrete = imported.f.get_concrete_function(
1028 imported.f.get_concrete_function(
1030 imported.f.get_concrete_function(
1041 root.f = func.get_concrete_function()
[all …]
/external/tensorflow/tensorflow/core/function/integration_test/
Dside_inputs_test.py80 self.assertLen(tf_f.get_concrete_function().graph.inputs, 1)
84 self.assertLen(tf_f.get_concrete_function().graph.inputs, 1)
183 self.assertLen(tf_f.get_concrete_function().graph.capture, 1)
/external/tensorflow/tensorflow/lite/python/authoring/
Dauthoring.py120 concrete_func = model.get_concrete_function(*args, **kwargs)
137 def get_concrete_function(self, *args, **kwargs): member in _Compatible
139 return self._get_func().get_concrete_function(*args, **kwargs)
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py52 init_func_concrete = _init_func.get_concrete_function()
63 self._init_func = _remote_init_func.get_concrete_function()
85 next_func_concrete = _next_func.get_concrete_function()
110 self._next_func = _remote_next_func.get_concrete_function()
130 finalize_func_concrete = _finalize_func.get_concrete_function()
143 self._finalize_func = _remote_finalize_func.get_concrete_function()
Dstructured_function.py205 _ = unused.get_concrete_function()
230 return wrapped_fn.get_concrete_function
244 return wrapped_fn.get_concrete_function
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsave_impl.py204 fn.get_concrete_function()
216 fn.get_concrete_function()
367 fn.get_concrete_function(*args, **kwargs)
369 fn.get_concrete_function(*args, **kwargs)
641 def get_concrete_function(self, *args, **kwargs): member in LayerCall
643 return self.wrapped_call.get_concrete_function(*args, **kwargs)
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants_test.py174 input_func = root.f.get_concrete_function()
218 to_save = root.f.get_concrete_function(input_data["x"])
241 input_func = root.f.get_concrete_function(input_data["x"])
258 input_func = root.f.get_concrete_function(input_data["x"])
275 to_save = root.f.get_concrete_function(input_data["x"])
313 input_func = root.add.get_concrete_function(input_data["x"])
543 input_func = root.f.get_concrete_function()
588 input_func = root.f.get_concrete_function(input_data["x"])
604 input_func = root.f.get_concrete_function(input_data["x"])
623 input_func = root.f.get_concrete_function(input_data["x"])
[all …]
Dfunction_def_to_graph_test.py119 lookup.get_concrete_function().add_to_graph()
236 fdef = fn.get_concrete_function(inp).function_def
252 fdef = fn.get_concrete_function(inp).function_def
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.types.experimental.-generic-function.pbtxt14 name: "get_concrete_function"

123456