Searched defs:memory_limit_bytes (Results 1 – 6 of 6) sorted by relevance
61 int64 memory_limit_bytes, RematerializationSizes* sizes) in HloRematerialization()
841 int64 memory_reduced, int64 memory_limit_bytes) { in RematerializationCost()864 const InstructionList& instruction_list, int64 memory_limit_bytes, in PickRematerializationCandidate()973 int64 memory_limit_bytes) { in RematerializeComputation()
144 StatusOr<bool> RunHloRematerialization(int64 memory_limit_bytes, in RunHloRematerialization()
587 def __init__(self, name, device_type, memory_limit_bytes, incarnation): argument602 def memory_limit_bytes(self): member in _DeviceAttributes
2160 size_t memory_limit_bytes) { in GetCudnnConvolutionForwardAlgo()2178 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardDataAlgo()2197 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardFilterAlgo()2353 auto memory_limit_bytes = in GetCudnnConvolutionForwardAlgorithm() local2402 auto memory_limit_bytes = in GetCudnnConvolutionBackwardDataAlgorithm() local2451 auto memory_limit_bytes = in GetCudnnConvolutionBackwardFilterAlgorithm() local
1116 std::vector<int64> memory_limit_bytes; in CreateDevices() local