Searched refs:call_def (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 117 NodeDef* call_def) { in RewriteSubgraph() argument
182 AddNodeAttr(EncapsulateXlaComputationsPass::kXlaClusterAttr, call_def->name(), in RewriteSubgraph()
183 call_def); in RewriteSubgraph()
184 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()
204 call_def->set_op(absl::StrCat(call_def->op(), "_", fingerprint)); in RewriteSubgraph()
|
| D | mark_for_compilation_pass.cc | 166 bool IsCompilableCall(const NodeDef& call_def,
219 bool IsCompilableCall(const NodeDef& call_def, in IsCompilableCall() argument
224 VLOG(2) << "Rejecting " << call_def.op() in IsCompilableCall()
230 Status status = InstantiateFunctionCall(call_def, *lib_runtime, &handle); in IsCompilableCall()
232 VLOG(2) << "Rejecting " << call_def.DebugString() in IsCompilableCall()
251 VLOG(2) << "Rejecting " << call_def.op() in IsCompilableCall()
285 VLOG(2) << "Rejecting " << call_def.op() << ": unsupported op " in IsCompilableCall()
|
| D | extract_outside_compilation_pass.cc | 1255 NodeDef call_def; in BuildHostGraphForFuncCallNode() local
1256 TF_RETURN_IF_ERROR(call_builder.Finalize(&call_def)); in BuildHostGraphForFuncCallNode()
1258 Node* call_node = host_graph.AddNode(call_def, &s); in BuildHostGraphForFuncCallNode()
|
| D | encapsulate_subgraphs_pass_test.cc | 806 NodeDef* call_def) { in TEST() argument
851 NodeDef* call_def) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 737 call_def = func_calls[uuid]
738 call_def.uuid = uuid
739 call_def.function_name = attr[OpHint.FUNCTION_NAME_ATTR].s
740 call_def.level = attr[OpHint.FUNCTION_LEVEL_ATTR].i
751 call_def.children_inputs_mappings = _json.loads(
765 put_operand(call_def.inputs, attr[OpHint.FUNCTION_INPUT_INDEX_ATTR].i,
768 put_operand(call_def.outputs, attr[OpHint.FUNCTION_OUTPUT_INDEX_ATTR].i,
774 call_def.params[a.replace("_tflite_attr_,", "")] = attr[a].tensor
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.h | 274 Status InstantiateFunctionCall(const NodeDef& call_def,
|
| D | function.cc | 1413 Status InstantiateFunctionCall(const NodeDef& call_def, in InstantiateFunctionCall() argument
1420 if (call_def.op() == "PartitionedCall" || in InstantiateFunctionCall()
1421 call_def.op() == "StatefulPartitionedCall") { in InstantiateFunctionCall()
1422 TF_RETURN_IF_ERROR(GetNodeAttr(call_def, "f", &func)); in InstantiateFunctionCall()
1426 func_name = &call_def.op(); in InstantiateFunctionCall()
1427 attrs = AttrSlice(call_def); in InstantiateFunctionCall()
|