Searched refs:bytes_limit (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_cpu_allocator.h | 58 stats_.bytes_limit = total_memory; in MklSmallSizeAllocator()
96 stats_.bytes_limit = 0; in ClearStats()
265 stats_.bytes_limit = std::max(s_stats->bytes_limit, l_stats->bytes_limit); in GetStats()
|
| D | mkl_cpu_allocator_test.cc | 31 EXPECT_EQ(stats->bytes_limit, 1000); in TEST()
41 EXPECT_EQ(stats->bytes_limit, max_mem_bytes); in TEST()
|
| D | bfc_allocator.cc | 50 stats_.bytes_limit = static_cast<int64>(total_memory); in BFCAllocator()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | allocator_stats.cc | 28 this->bytes_limit ? *this->bytes_limit : 0, this->bytes_in_use, in DebugString()
|
| D | allocator_stats.h | 37 absl::optional<int64> bytes_limit; member
|
| /external/tensorflow/tensorflow/contrib/memory_stats/kernels/ |
| D | memory_stats_ops.cc | 76 return allocator_stats.bytes_limit ? *allocator_stats.bytes_limit : -1; in ExtractAllocatorStats()
|
| /external/tensorflow/tensorflow/contrib/memory_stats/python/kernel_tests/ |
| D | memory_stats_ops_test.py | 40 bytes_limit = sess.run(memory_stats_ops.BytesLimit())
41 self.assertLess(0, bytes_limit)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 36 this->bytes_limit ? *this->bytes_limit : 0, this->bytes_in_use, in DebugString()
|
| D | allocator.h | 68 absl::optional<int64> bytes_limit; member
|
| /external/tensorflow/tensorflow/contrib/memory_stats/python/ops/ |
| D | memory_stats_ops.py | 36 return gen_memory_stats_ops.bytes_limit()
|
| /external/protobuf/src/google/protobuf/ |
| D | wire_format_lite_inl.h | 365 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/compiler/jit/ |
| D | xla_device_context.cc | 67 tf_stats.bytes_limit = se_stats->bytes_limit; in GetStats()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.cc | 26 stats_.bytes_limit = in SYCLAllocator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 248 if (stats && *stats->bytes_limit && in ComputeAsync()
249 stats->bytes_in_use > (*stats->bytes_limit * kOccupancy)) { in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 1217 int64 bytes_limit = stats->bytes_limit ? *stats->bytes_limit : 0; in CreateGPUDevice() local
1219 options, device_name, static_cast<Bytes>(bytes_limit), dev_locality, in CreateGPUDevice()
1223 << (bytes_limit >> 20) << " MB memory) -> physical GPU (" in CreateGPUDevice()
|