Searched defs:memory_limit_bytes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.cc | 512 int64_t memory_limit_bytes) { in RematerializationCost()
1325 int64_t memory_limit_bytes) { in RematerializationCost()
1379 const InstructionList& instruction_list, int64_t memory_limit_bytes, in PickRematerializationCandidates()
1761 InstructionList* instruction_list, int64_t memory_limit_bytes, in RematerializeBestBlock()
1844 int64_t memory_limit_bytes, int64_t min_remat_size) { in RematerializeComputation()
|
| D | hlo_rematerialization_test.cc | 43 StatusOr<bool> RunHloRematerialization(int64_t memory_limit_bytes, in RunHloRematerialization()
600 StatusOr<bool> RunHloRematerialization(int64_t memory_limit_bytes, in RunHloRematerialization()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_factory.cc | 190 std::vector<int64> memory_limit_bytes; in CreateDevices() local
|
| /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 | 2707 size_t memory_limit_bytes) { in GetCudnnConvolutionForwardAlgo()
2748 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardDataAlgo()
2791 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardFilterAlgo()
3001 auto memory_limit_bytes = in GetCudnnConvolutionForwardAlgorithm() local
3063 auto memory_limit_bytes = in GetCudnnConvolutionBackwardDataAlgorithm() local
3124 auto memory_limit_bytes = in GetCudnnConvolutionBackwardFilterAlgorithm() local
3844 auto memory_limit_bytes = in GetFirstWorkingExecutionPlan() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 1384 std::vector<int64> memory_limit_bytes; in CreateDevices() local
|