Searched refs:ExtendNodeAttr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer_test.cc | 224 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {0, 2}, &nd); in TEST_F()
225 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {6, 7}, &nd); in TEST_F()
226 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", {2, 3}, &nd); in TEST_F()
|
| D | scoped_allocator_optimizer.h | 70 static void ExtendNodeAttr(StringPiece name, const std::vector<int32>& values,
|
| D | scoped_allocator_optimizer.cc | 201 void ScopedAllocatorOptimizer::ExtendNodeAttr(StringPiece name, in ExtendNodeAttr() function in tensorflow::grappler::ScopedAllocatorOptimizer
342 ScopedAllocatorOptimizer::ExtendNodeAttr("_scoped_allocator", in ConstructScopedAllocatorNode()
|