Home
last modified time | relevance | path

Searched refs:memory_limit_bytes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.h61 int64 memory_limit_bytes, RematerializationSizes* sizes) in HloRematerialization() argument
63 memory_limit_bytes_(memory_limit_bytes), in HloRematerialization()
84 int64 memory_limit_bytes);
Dhlo_rematerialization.cc841 int64 memory_reduced, int64 memory_limit_bytes) { in RematerializationCost() argument
854 return memory_limit_bytes / memory_reduced; in RematerializationCost()
864 const InstructionList& instruction_list, int64 memory_limit_bytes, in PickRematerializationCandidate() argument
918 memory_reduced, memory_limit_bytes); in PickRematerializationCandidate()
973 int64 memory_limit_bytes) { in RematerializeComputation() argument
975 << " with limit " << HumanReadableNumBytes(memory_limit_bytes); in RematerializeComputation()
1024 while (memory_tracker.memory_usage() + callee_usage > memory_limit_bytes) { in RematerializeComputation()
1029 << ", limit is " << HumanReadableNumBytes(memory_limit_bytes); in RematerializeComputation()
1032 memory_tracker, instruction_list, memory_limit_bytes, &remat_able); in RematerializeComputation()
1134 memory_tracker.memory_usage() + callee_usage > memory_limit_bytes) { in RematerializeComputation()
[all …]
Dhlo_rematerialization_test.cc144 StatusOr<bool> RunHloRematerialization(int64 memory_limit_bytes, in RunHloRematerialization() argument
151 HloRematerialization remat(ByteSizeOf, memory_limit_bytes, in RunHloRematerialization()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1116 std::vector<int64> memory_limit_bytes; in CreateDevices() local
1124 memory_limit_bytes.push_back(single_virtual_device_memory_limit); in CreateDevices()
1128 std::back_inserter(memory_limit_bytes), [](float mb) { in CreateDevices()
1132 while (next_tf_gpu_id < memory_limit_bytes.size()) { in CreateDevices()
1146 CHECK_EQ(next_tf_gpu_id, memory_limit_bytes.size()); in CreateDevices()
1149 int64 bytes = memory_limit_bytes[di]; in CreateDevices()
/external/tensorflow/tensorflow/python/client/
Dsession.py587 def __init__(self, name, device_type, memory_limit_bytes, incarnation): argument
590 self._memory_limit_bytes = memory_limit_bytes
602 def memory_limit_bytes(self): member in _DeviceAttributes
613 self.memory_limit_bytes,
Dsession_test.py1992 self.assertEqual(1337, attrs.memory_limit_bytes)
2002 self.assertEqual(1337, attrs.memory_limit_bytes)
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2160 size_t memory_limit_bytes) { in GetCudnnConvolutionForwardAlgo() argument
2167 output_nd.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionForwardAlgo()
2178 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardDataAlgo() argument
2186 input_nd.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionBackwardDataAlgo()
2197 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardFilterAlgo() argument
2205 filter.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionBackwardFilterAlgo()
2353 auto memory_limit_bytes = in GetCudnnConvolutionForwardAlgorithm() local
2360 specify_workspace_limit, memory_limit_bytes)); in GetCudnnConvolutionForwardAlgorithm()
2402 auto memory_limit_bytes = in GetCudnnConvolutionBackwardDataAlgorithm() local
2409 specify_workspace_limit, memory_limit_bytes)); in GetCudnnConvolutionBackwardDataAlgorithm()
[all …]