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.h84 const ShapeSizeFunction& size_function, int64 memory_limit_bytes,
91 memory_limit_bytes_(memory_limit_bytes), in size_function_()
119 int64 memory_limit_bytes,
Dhlo_rematerialization.cc510 int64 memory_reduced, int64 memory_limit_bytes) { in RematerializationCost() argument
530 return memory_limit_bytes / memory_reduced; in RematerializationCost()
565 const InstructionList& instruction_list, int64 memory_limit_bytes,
1319 int64 memory_reduced, int64 memory_limit_bytes) { in RematerializationCost() argument
1332 return memory_limit_bytes / memory_reduced; in RematerializationCost()
1373 const InstructionList& instruction_list, int64 memory_limit_bytes, in PickRematerializationCandidates() argument
1422 const int64 cost = memory_limit_bytes / memory_reduced; in PickRematerializationCandidates()
1471 RematerializationCost(block, memory_reduced, memory_limit_bytes); in PickRematerializationCandidates()
1751 InstructionList* instruction_list, int64 memory_limit_bytes, in RematerializeBestBlock() argument
1760 *instruction_list, memory_limit_bytes, rematerializable_map, in RematerializeBestBlock()
[all …]
Dhlo_rematerialization_test.cc43 StatusOr<bool> RunHloRematerialization(int64 memory_limit_bytes, in RunHloRematerialization() argument
52 ByteSizeOf, memory_limit_bytes, in RunHloRematerialization()
600 StatusOr<bool> RunHloRematerialization(int64 memory_limit_bytes, in RunHloRematerialization() argument
605 ShapeSizePadMinorTo64, memory_limit_bytes, in RunHloRematerialization()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2415 size_t memory_limit_bytes) { in GetCudnnConvolutionForwardAlgo() argument
2426 size_t mem_limit = specify_workspace_limit ? memory_limit_bytes : 0ULL; in GetCudnnConvolutionForwardAlgo()
2444 output_nd.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionForwardAlgo()
2456 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardDataAlgo() argument
2467 size_t mem_limit = specify_workspace_limit ? memory_limit_bytes : 0ULL; in GetCudnnConvolutionBackwardDataAlgo()
2487 input_nd.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionBackwardDataAlgo()
2499 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardFilterAlgo() argument
2509 size_t mem_limit = specify_workspace_limit ? memory_limit_bytes : 0ULL; in GetCudnnConvolutionBackwardFilterAlgo()
2529 filter.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionBackwardFilterAlgo()
2709 auto memory_limit_bytes = in GetCudnnConvolutionForwardAlgorithm() local
[all …]
/external/tensorflow/tensorflow/python/client/
Dsession.py606 def __init__(self, name, device_type, memory_limit_bytes, incarnation): argument
609 self._memory_limit_bytes = memory_limit_bytes
621 def memory_limit_bytes(self): member in _DeviceAttributes
632 self.memory_limit_bytes,
Dsession_test.py2035 self.assertEqual(1337, attrs.memory_limit_bytes)
2045 self.assertEqual(1337, attrs.memory_limit_bytes)
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1334 std::vector<int64> memory_limit_bytes; in CreateDevices() local
1342 memory_limit_bytes.push_back(single_virtual_device_memory_limit); in CreateDevices()
1346 std::back_inserter(memory_limit_bytes), [](float mb) { in CreateDevices()
1350 while (next_tf_gpu_id < memory_limit_bytes.size()) { in CreateDevices()
1364 CHECK_EQ(next_tf_gpu_id, memory_limit_bytes.size()); in CreateDevices()
1367 int64 bytes = memory_limit_bytes[di]; in CreateDevices()