Searched defs:memory_limit_bytes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.cc | 510 int64 memory_reduced, int64 memory_limit_bytes) { in RematerializationCost()
1319 int64 memory_reduced, int64 memory_limit_bytes) { in RematerializationCost()
1373 const InstructionList& instruction_list, int64 memory_limit_bytes, in PickRematerializationCandidates()
1751 InstructionList* instruction_list, int64 memory_limit_bytes, in RematerializeBestBlock()
1832 int64 memory_limit_bytes, int64 min_remat_size) { in RematerializeComputation()
|
| D | hlo_rematerialization_test.cc | 43 StatusOr<bool> RunHloRematerialization(int64 memory_limit_bytes, in RunHloRematerialization()
600 StatusOr<bool> RunHloRematerialization(int64 memory_limit_bytes, in RunHloRematerialization()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session.py | 606 def __init__(self, name, device_type, memory_limit_bytes, incarnation): argument
621 def memory_limit_bytes(self): member in _DeviceAttributes
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2415 size_t memory_limit_bytes) { in GetCudnnConvolutionForwardAlgo()
2456 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardDataAlgo()
2499 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardFilterAlgo()
2709 auto memory_limit_bytes = in GetCudnnConvolutionForwardAlgorithm() local
2771 auto memory_limit_bytes = in GetCudnnConvolutionBackwardDataAlgorithm() local
2832 auto memory_limit_bytes = in GetCudnnConvolutionBackwardFilterAlgorithm() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 1334 std::vector<int64> memory_limit_bytes; in CreateDevices() local
|