Searched refs:bytes_limit (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | allocator_stats.cc | 31 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/core/common_runtime/ |
| D | mkl_cpu_allocator.h | 58 stats_.bytes_limit = total_memory; in MklSmallSizeAllocator()
96 stats_.bytes_limit = 0; in ClearStats()
267 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 | 60 stats_.bytes_limit = static_cast<int64>(total_memory); in BFCAllocator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 42 static_cast<long long>(this->bytes_limit ? *this->bytes_limit : 0), in DebugString()
|
| D | allocator.h | 156 absl::optional<int64> bytes_limit; member
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 68 tf_stats.bytes_limit = se_stats->bytes_limit; in GetStats()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 249 if (stats && *stats->bytes_limit && in ComputeAsync()
250 stats->bytes_in_use > (*stats->bytes_limit * kOccupancy)) { in ComputeAsync()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.cc | 234 stats.bytes_limit = c_stats.bytes_limit; in GetAllocatorStats()
|
| D | c_api_decl.h | 89 int64_t bytes_limit; member
|
| /external/protobuf/src/google/protobuf/ |
| D | wire_format_lite.h | 1177 int64 bytes_limit = input->BytesUntilTotalBytesLimit(); in ReadPackedFixedSizePrimitive() local
1178 if (bytes_limit == -1) { in ReadPackedFixedSizePrimitive()
1179 bytes_limit = input->BytesUntilLimit(); in ReadPackedFixedSizePrimitive()
1182 bytes_limit = in ReadPackedFixedSizePrimitive()
1183 (std::min)(bytes_limit, static_cast<int64>(input->BytesUntilLimit())); in ReadPackedFixedSizePrimitive()
1185 if (bytes_limit >= new_bytes) { in ReadPackedFixedSizePrimitive()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 1451 int64 bytes_limit = stats->bytes_limit ? *stats->bytes_limit : 0; in CreateGPUDevice() local
1453 options, device_name, static_cast<Bytes>(bytes_limit), dev_locality, in CreateGPUDevice()
1457 << (bytes_limit >> 20) << " MB memory) -> physical GPU (" in CreateGPUDevice()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.h | 115 int64_t bytes_limit; member
|
| D | stream_executor.cc | 322 stats.bytes_limit = c_stats.bytes_limit; in GetAllocatorStats()
|