Searched refs:sa_opti (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 215 Status MaybeRewriteInput(ScopedAllocatorOptimizer* sa_opti, in MaybeRewriteInput() argument
222 (sa_opti->repeated_outputs().find(edge_name) != in MaybeRewriteInput()
223 sa_opti->repeated_outputs().end()); in MaybeRewriteInput()
232 LOG_WARNING_AND_RETURN_IF_ERROR(sa_opti->NewIdentityId(&unique_id)); in MaybeRewriteInput()
258 Status GetInputs(ScopedAllocatorOptimizer* sa_opti, int64 invocation_count, in GetInputs() argument
283 sa_opti, invocation_count, graph, node_map, dtype, inode, in GetInputs()
446 Status AnalyzeInputs(ScopedAllocatorOptimizer* sa_opti, in AnalyzeInputs() argument
457 GetInputs(sa_opti, invocation_count, graph, *graph_properties_, in AnalyzeInputs()
458 sa_opti->node_map(), ops, *dtype, inputs)); in AnalyzeInputs()
511 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode() argument
[all …]
|