Searched defs:memory_limit_bytes (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_rematerialization.cc | 515 int64_t memory_limit_bytes) { in RematerializationCost()
1333 int64_t memory_limit_bytes) { in RematerializationCost()
1387 const InstructionList& instruction_list, int64_t memory_limit_bytes, in PickRematerializationCandidates()
1796 InstructionList* instruction_list, int64_t memory_limit_bytes, in RematerializeBestBlock()
1895 int64_t memory_limit_bytes, int64_t min_remat_size, in RematerializeComputation()
|
| H A D | hlo_rematerialization_test.cc | 43 StatusOr<bool> RunHloRematerialization(int64_t memory_limit_bytes, in RunHloRematerialization()
603 StatusOr<bool> RunHloRematerialization(int64_t memory_limit_bytes, in RunHloRematerialization()
|
| H A D | hlo_rematerialization.h | 132 int64_t memory_limit_bytes, in RematerializeComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | session.py | 603 def __init__(self, name, device_type, memory_limit_bytes, incarnation): argument
618 def memory_limit_bytes(self): member in _DeviceAttributes
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_factory.cc | 191 std::vector<int64_t> memory_limit_bytes; in CreateDevices() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 2767 size_t memory_limit_bytes) { in GetCudnnConvolutionForwardAlgo()
2808 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardDataAlgo()
2851 size_t memory_limit_bytes) { in GetCudnnConvolutionBackwardFilterAlgo()
3074 auto memory_limit_bytes = in GetCudnnConvolutionForwardAlgorithm() local
3136 auto memory_limit_bytes = in GetCudnnConvolutionBackwardDataAlgorithm() local
3197 auto memory_limit_bytes = in GetCudnnConvolutionBackwardFilterAlgorithm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device.cc | 1456 int64_t memory_limit_bytes; in CreateDevices() member
|