Searched defs:while_node (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 68 Node* while_node; in TEST() local
190 Node* while_node; in TEST() local
295 Node* while_node; in TEST() local
418 Node* while_node; in TEST() local
533 Node* while_node; in TEST() local
|
| D | lower_functional_ops_test.cc | 157 Node* while_node; in TEST() local
|
| D | executor_test.cc | 613 Node* while_node; in BM_WhileLoopHelper() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | rearrange_function_argument_pass_test.cc | 154 const Node *while_node = node_name_index.at("while"); in TEST() local
|
| D | compilability_check_util.cc | 277 const Node& while_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableWhile()
|
| D | extract_outside_compilation_pass.cc | 1477 Node* while_node) { in AddSendLoopPredToLoopCond()
1736 Node* while_node = host_graph.AddNode(while_def, &s); in BuildHostGraphForWhileNode() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1763 Node* while_node, FunctionLibraryDefinition* fld) { in FindArgsToLiftForWhileNode()
2043 void RemoveOutputIdentityNodesForWhileV2(Graph* g, Node* while_node) { in RemoveOutputIdentityNodesForWhileV2()
2071 const absl::flat_hash_set<int>& args_to_lift, Graph* g, Node* while_node) { in ReplaceOutputEdgesWithInputEdgeSourceForWhile()
2124 Graph* g, Node* while_node, FunctionLibraryDefinition* fld, in LiftOutsideCompilationOnlyArgsFromWhileNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 407 Node* while_node = node_name_index["while"]; in TEST() local
|
| D | tf2xla_util.cc | 251 Status PropagateConstIntoWhileNode(Graph* g, Node* while_node, in PropagateConstIntoWhileNode()
|