Searched refs:node_rewritten (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 279 Graph* g, Node* n, FunctionLibraryDefinition* fld, bool* node_rewritten) { in MaybeRewriteWhileNode() argument
289 *node_rewritten = false; in MaybeRewriteWhileNode()
293 *node_rewritten = true; in MaybeRewriteWhileNode()
383 Graph* g, Node* n, FunctionLibraryDefinition* fld, bool* node_rewritten, in MaybeRewriteIfNode() argument
400 *node_rewritten = false; in MaybeRewriteIfNode()
404 *node_rewritten = true; in MaybeRewriteIfNode()
538 bool node_rewritten; in RearrangeFunctionArguments() local
540 &node_rewritten)); in RearrangeFunctionArguments()
542 bool node_rewritten; in RearrangeFunctionArguments() local
544 &node_rewritten, global_fld)); in RearrangeFunctionArguments()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 2474 bool node_rewritten = false; in LiftOutsideCompilationOnlyArgs() local
2476 g, n, fld, lifted_arg_count, &node_rewritten)); in LiftOutsideCompilationOnlyArgs()
2477 *rewritten = *rewritten || node_rewritten; in LiftOutsideCompilationOnlyArgs()
2481 bool node_rewritten = false; in LiftOutsideCompilationOnlyArgs() local
2483 g, n, fld, lifted_arg_count, &node_rewritten)); in LiftOutsideCompilationOnlyArgs()
2484 *rewritten = *rewritten || node_rewritten; in LiftOutsideCompilationOnlyArgs()
2488 bool node_rewritten = false; in LiftOutsideCompilationOnlyArgs() local
2490 g, n, flr, fld, lifted_arg_count, &node_rewritten)); in LiftOutsideCompilationOnlyArgs()
2491 *rewritten = *rewritten || node_rewritten; in LiftOutsideCompilationOnlyArgs()
|