Searched refs:scoped_allocator_opts (Results 1 – 10 of 10) sorted by relevance
188 del rewrite_options.scoped_allocator_opts.enable_op[:]189 rewrite_options.scoped_allocator_opts.enable_op.append('CollectiveReduce')249 del rewrite_options.scoped_allocator_opts.enable_op[:]250 rewrite_options.scoped_allocator_opts.enable_op.append('CollectiveReduce')543 del rewrite_options.scoped_allocator_opts.enable_op[:]544 rewrite_options.scoped_allocator_opts.enable_op.append('CollectiveReduce')
44 del rewrite_options.scoped_allocator_opts.enable_op[:]45 rewrite_options.scoped_allocator_opts.enable_op.append('CollectiveReduce')
1007 const ::tensorflow::ScopedAllocatorOptions& scoped_allocator_opts() const; in Swap()1010 …d set_allocated_scoped_allocator_opts(::tensorflow::ScopedAllocatorOptions* scoped_allocator_opts); in Swap()1016 ::tensorflow::ScopedAllocatorOptions* scoped_allocator_opts); in Swap()2304 inline const ::tensorflow::ScopedAllocatorOptions& RewriterConfig::scoped_allocator_opts() const { in scoped_allocator_opts() function2309 ::tensorflow::ScopedAllocatorOptions* scoped_allocator_opts) { in unsafe_arena_set_allocated_scoped_allocator_opts() argument2313 _impl_.scoped_allocator_opts_ = scoped_allocator_opts; in unsafe_arena_set_allocated_scoped_allocator_opts()2351 …:set_allocated_scoped_allocator_opts(::tensorflow::ScopedAllocatorOptions* scoped_allocator_opts) { in set_allocated_scoped_allocator_opts() argument2356 if (scoped_allocator_opts) { in set_allocated_scoped_allocator_opts()2358 ::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(scoped_allocator_opts); in set_allocated_scoped_allocator_opts()2360 scoped_allocator_opts = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage( in set_allocated_scoped_allocator_opts()[all …]
1100 …static const ::tensorflow::ScopedAllocatorOptions& scoped_allocator_opts(const RewriterConfig* msg…1110 RewriterConfig::_Internal::scoped_allocator_opts(const RewriterConfig* msg) { in scoped_allocator_opts() function in tensorflow::RewriterConfig::_Internal1817 InternalWriteMessage(16, _Internal::scoped_allocator_opts(this), in _InternalSerialize()1818 _Internal::scoped_allocator_opts(this).GetCachedSize(), target, stream); in _InternalSerialize()
757 del rewrite_options.scoped_allocator_opts.enable_op[:]758 rewrite_options.scoped_allocator_opts.enable_op.append("CollectiveReduce")
406 rewrite_options.scoped_allocator_opts.enable_op.append('to_be_removed')422 new_rewrite_options.scoped_allocator_opts.enable_op)
256 cfg_.scoped_allocator_opts())); in MakeNewOptimizer()484 cfg_.scoped_allocator_optimization(), cfg_.scoped_allocator_opts())); in InitializeOptimizers()
208 ScopedAllocatorOptions scoped_allocator_opts = 16; field
1163 del rewrite_options.scoped_allocator_opts.enable_op[:]1165 rewrite_options.scoped_allocator_opts.enable_op.append(op)
857 new_rewrite_options.scoped_allocator_opts.enable_op)