Home
last modified time | relevance | path

Searched refs:outside_compilation_node (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc480 Node* outside_compilation_node = pair.second; in UpdateTypesAttribute() local
482 TF_RET_CHECK(outside_compilation_node->IsIdentity() || in UpdateTypesAttribute()
483 outside_compilation_node->type_string() == "Placeholder"); in UpdateTypesAttribute()
484 if (outside_compilation_node->IsIdentity()) { in UpdateTypesAttribute()
486 GetNodeAttr(outside_compilation_node->def(), "T", &data_type)); in UpdateTypesAttribute()
489 GetNodeAttr(outside_compilation_node->def(), "dtype", &data_type)); in UpdateTypesAttribute()
506 Node* outside_compilation_node = in AddEdgesFromOutsideCompilationNodes() local
508 g->AddEdge(outside_compilation_node, 0, n, i + arg_to_input_edge_offset); in AddEdgesFromOutsideCompilationNodes()
847 Node* outside_compilation_node = pair.second; in PostprocessLiftedArgsForCall() local
849 TF_RET_CHECK(outside_compilation_node->IsIdentity() || in PostprocessLiftedArgsForCall()
[all …]