Home
last modified time | relevance | path

Searched refs:bytes_limit (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_cpu_allocator_test.cc32 EXPECT_EQ(stats.bytes_limit, 1000); in TEST()
42 EXPECT_EQ(stats.bytes_limit, max_mem_bytes); in TEST()
Dbfc_allocator.cc49 stats_.bytes_limit = static_cast<int64>(total_memory); in BFCAllocator()
/external/tensorflow/tensorflow/contrib/memory_stats/python/kernel_tests/
Dmemory_stats_ops_test.py40 bytes_limit = sess.run(memory_stats_ops.BytesLimit())
41 self.assertLess(0, bytes_limit)
/external/tensorflow/tensorflow/core/framework/
Dallocator.cc33 this->bytes_limit = 0; in Clear()
43 this->bytes_limit, this->bytes_in_use, this->max_bytes_in_use, in DebugString()
Dallocator.h58 int64 bytes_limit; member
/external/protobuf/src/google/protobuf/
Dwire_format_lite_inl.h365 int64 bytes_limit = input->BytesUntilTotalBytesLimit(); in ReadPackedFixedSizePrimitive() local
366 if (bytes_limit == -1) { in ReadPackedFixedSizePrimitive()
367 bytes_limit = input->BytesUntilLimit(); in ReadPackedFixedSizePrimitive()
369 bytes_limit = in ReadPackedFixedSizePrimitive()
370 std::min(bytes_limit, static_cast<int64>(input->BytesUntilLimit())); in ReadPackedFixedSizePrimitive()
372 if (bytes_limit >= new_bytes) { in ReadPackedFixedSizePrimitive()
/external/tensorflow/tensorflow/contrib/memory_stats/python/ops/
Dmemory_stats_ops.py36 return gen_memory_stats_ops.bytes_limit()
/external/tensorflow/tensorflow/contrib/memory_stats/kernels/
Dmemory_stats_ops.cc74 return allocator_stats.bytes_limit; in ExtractAllocatorStats()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_allocator.cc26 stats_.bytes_limit = in SYCLAllocator()
/external/tensorflow/tensorflow/core/kernels/
Dstack_ops.cc288 if (stats.bytes_in_use > (stats.bytes_limit * kOccupancy)) { in ComputeAsync()