Searched refs:memory_limit_mb (Results 1 – 8 of 8) sorted by relevance
102 EXPECT_EQ(f->memory_limit_mb, 666); in VerifyOptions1()127 EXPECT_EQ(f->memory_limit_mb, 456); in VerifyOptions2()
102 name: "memory_limit_mb"
71 const std::vector<std::vector<float>>& memory_limit_mb = {}) { in MakeSessionOptions() argument79 for (const auto& v : memory_limit_mb) {
1126 const auto& memory_limit_mb = virtual_devices.Get(i).memory_limit_mb(); in CreateDevices() local1127 std::transform(memory_limit_mb.begin(), memory_limit_mb.end(), in CreateDevices()
57 int, memory_limit_mb, 0,
189 if (flags()->memory_limit_mb > 0) { in BackgroundThread()191 uptr limit = uptr(flags()->memory_limit_mb) << 20; in BackgroundThread()
94 memory_limit_mb=i) for i in self._mem_limits_mb
110 repeated float memory_limit_mb = 1; field