/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()
|