| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_utils.cc | 262 Status NameAndAttrsFromFunctionCall(const NodeDef& call_def, in NameAndAttrsFromFunctionCall() argument
264 if (call_def.op() == "PartitionedCall" || in NameAndAttrsFromFunctionCall()
265 call_def.op() == "StatefulPartitionedCall") { in NameAndAttrsFromFunctionCall()
266 TF_RETURN_IF_ERROR(GetNodeAttr(call_def, "f", function)); in NameAndAttrsFromFunctionCall()
268 function->set_name(call_def.op()); in NameAndAttrsFromFunctionCall()
269 *function->mutable_attr() = call_def.attr(); in NameAndAttrsFromFunctionCall()
274 Status InstantiateFunctionCall(const NodeDef& call_def, in InstantiateFunctionCall() argument
278 TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(call_def, &function)); in InstantiateFunctionCall()
|
| D | function_utils.h | 52 Status NameAndAttrsFromFunctionCall(const NodeDef& call_def,
94 Status InstantiateFunctionCall(const NodeDef& call_def,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.h | 171 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime,
184 bool IsCompilableCall(const NodeDef& call_def, in IsCompilableCall() argument
187 stack_trace.emplace_back(StackFrameView{call_def.name(), ""}); in IsCompilableCall()
188 return IsCompilableCall(call_def, lib_runtime, &stack_trace); in IsCompilableCall()
209 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime,
|
| D | encapsulate_xla_computations_pass.cc | 113 NodeDef* call_def) { in RewriteSubgraph() argument
178 AddNodeAttr(kXlaClusterIdAttr, call_def->name(), call_def); in RewriteSubgraph()
179 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()
199 call_def->set_op(absl::StrCat(call_def->op(), "_", fingerprint)); in RewriteSubgraph()
|
| D | compilability_check_util.cc | 169 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime, in FindUncompilableNodes() argument
182 stack_trace.emplace_back(StackFrameView{call_def.name(), "", nullptr}); in FindUncompilableNodes()
185 IsCompilableCall(call_def, lib_runtime, &stack_trace, in FindUncompilableNodes()
327 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime, in IsCompilableCall() argument
336 VLOG(2) << "Rejecting " << call_def.op() << ": " << uncompilable_reason in IsCompilableCall()
344 s = NameAndAttrsFromFunctionCall(call_def, &function); in IsCompilableCall()
354 VLOG(2) << "Rejecting " << call_def.DebugString() << ": " in IsCompilableCall()
|
| D | extract_outside_compilation_pass.cc | 1783 NodeDef call_def; in BuildHostGraphForFuncCallNode() local
1784 TF_RETURN_IF_ERROR(call_builder.Finalize(&call_def)); in BuildHostGraphForFuncCallNode()
1786 Node* call_node = host_graph.AddNode(call_def, &s); in BuildHostGraphForFuncCallNode()
|
| D | encapsulate_subgraphs_pass_test.cc | 799 NodeDef* call_def) { in TEST() argument
844 NodeDef* call_def) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 758 call_def = func_calls[uuid]
759 call_def.uuid = uuid
760 call_def.function_name = attr[OpHint.FUNCTION_NAME_ATTR].s
761 call_def.level = attr[OpHint.FUNCTION_LEVEL_ATTR].i
774 call_def.children_inputs_mappings = _json.loads(
788 put_operand(call_def.inputs, attr[OpHint.FUNCTION_INPUT_INDEX_ATTR].i,
791 put_operand(call_def.outputs, attr[OpHint.FUNCTION_OUTPUT_INDEX_ATTR].i,
797 call_def.params[a.replace("_tflite_attr_,", "")] = attr[a].tensor
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 80 NodeDef* call_def) { in RewriteSubgraph() argument
170 call_def->mutable_attr()->erase("num_cores_per_replica"); in RewriteSubgraph()
171 AddNodeAttr("num_cores_per_replica", num_cores_per_replica, call_def); in RewriteSubgraph()
174 call_def->mutable_attr()->insert(attr); in RewriteSubgraph()
177 MergeDebugInfo(NodeDebugInfo(metadata_node->def()), call_def); in RewriteSubgraph()
241 AddNodeAttr(kTPUReplicateAttr, call_def->name(), call_def); in RewriteSubgraph()
242 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()
244 call_def); in RewriteSubgraph()
264 call_def->set_op(strings::StrCat(call_def->op(), "_", fingerprint)); in RewriteSubgraph()
|