Searched refs:sa_opti (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 279 Status AnalyzeInputs(ScopedAllocatorOptimizer* sa_opti, NodeMap* node_map, in AnalyzeInputs() argument
289 GetInputs(sa_opti->node_map(), ops, *dtype, inputs)); in AnalyzeInputs()
314 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode() argument
568 Status Rewrite(ScopedAllocatorOptimizer* sa_opti, int64 invocation_count, in Rewrite() argument
580 NodeMap* node_map = sa_opti->node_map(); in Rewrite()
594 TF_RETURN_IF_ERROR(AnalyzeInputs(sa_opti, node_map, ops, op_instance_names, in Rewrite()
598 int sa_id = sa_opti->NewScopedAllocatorId(input_shapes.size()); in Rewrite()
602 sa_opti, graph, node_map, ops, device_name, dtype, sa_id, sa_name, in Rewrite()
|