- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for _input_hostmem (0.16 sec)
-
tensorflow/compiler/jit/force_xla_constants_on_host_pass.h
#include "tensorflow/core/common_runtime/optimization_registry.h" namespace tensorflow { // An optimization pass which marks the constants which have to be resolved for // XLA compilation with `_input_hostmem`. class ForceXlaConstantsOnHostPass : public GraphOptimizationPass { public: ForceXlaConstantsOnHostPass() = default; Status Run(const GraphOptimizationPassOptions& options) override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 01 18:34:32 UTC 2020 - 1.4K bytes - Viewed (0) -
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)