- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetTestGraph (0.1 sec)
-
tensorflow/compiler/jit/tests/device_compiler_test_helper.h
} JitCompilationListener* listener() const { return listener_; } // Returns a test graph that will split into two XLA clusters (due to a node // with _XlaCompile = false). GraphDef GetTestGraph(const PartialTensorShape& input_shape); // Runs the graph using specified batch size both with and without XLA JIT // compilation. Returns an error if the results between the two do not match.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
for (const auto& attr : attrs) node.mutable_attr()->insert({attr.first, attr.second.proto}); return node; } } // namespace GraphDef DeviceCompilerSerializeTest::GetTestGraph( const PartialTensorShape& input_shape) { FunctionDef make_test_fn = FunctionDefHelper::Define( "TestFn", {"a:float", "b:float", "c:float"}, {"m:float"}, {}, {{{"d"}, "Add", {"a", "b"}, {{"T", DT_FLOAT}}},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0)