Searched refs:sub_func (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | sub.h | 98 auto sub_func = [&](int indexes[N]) { in BroadcastSubSlow()
105 NDOpsHelper<N>(output_desc, sub_func); in BroadcastSubSlow()
138 auto sub_func = [&](int indexes[N]) { in BroadcastSubSlow()
162 NDOpsHelper<N>(output_desc, sub_func); in BroadcastSubSlow()
195 auto sub_func = [&](int indexes[N]) { in BroadcastSubSlow()
202 NDOpsHelper<N>(output_desc, sub_func); in BroadcastSubSlow()
232 auto sub_func = [&](int indexes[N]) { in BroadcastSubSlow()
256 NDOpsHelper<N>(output_desc, sub_func); in BroadcastSubSlow()
288 auto sub_func = [&](int indexes[N]) { in BroadcastSubSlow()
295 NDOpsHelper<N>(output_desc, sub_func); in BroadcastSubSlow()
[all …]
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 384 sub_func = original_graph._get_function(func) # pylint: disable=protected-access
385 … if isinstance(sub_func, function._EagerDefinedFunction): # pylint: disable=protected-access
387 (func, sub_func.graph.get_operation_by_name(name)))
|
| D | lite_v2_test.py | 135 sub_func = root.sub.get_concrete_function(input_data)
139 [add_func, sub_func])
1053 sub_func = root.sub.get_concrete_function(input_data)
1056 save(root, save_dir, {'add': add_func, 'sub': sub_func})
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | py_func_test.py | 103 def sub_func(x, y): function
109 z = self.evaluate(script_ops.py_func(sub_func, [x, y], dtype))
|