- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _input_hostmem (0.1 sec)
-
tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc
for (Node* node : graph->nodes()) { if (CanCreateXlaKernel(node->def())) { EXPECT_FALSE(found); found = true; std::vector<int32> hostmem_attr; EXPECT_TRUE(TryGetNodeAttr(node->def(), "_input_hostmem", &hostmem_attr)); EXPECT_EQ(hostmem_attr.size(), 1); EXPECT_EQ(hostmem_attr[0], 1); } } EXPECT_TRUE(found); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.1K bytes - Viewed (0)