- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for custom_gradient_test (0.2 sec)
-
tensorflow/c/experimental/gradients/custom_gradient_test.cc
EXPECT_EQ(*result_value, 1.0); outputs[0]->Unref(); TF_DeleteTensor(result_tensor); result_tensor = nullptr; } INSTANTIATE_TEST_SUITE_P( CustomGradientTest, CustomGradientTest, ::testing::Combine(::testing::Values("graphdef", "mlir"), /*tfrt*/ ::testing::Values(false), /*executing_eagerly*/ ::testing::Values(true, false)));
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/BUILD
"//tensorflow/c/eager:abstract_context", "//tensorflow/c/eager:gradients_internal", ], ) tf_cuda_cc_test( name = "custom_gradient_test", size = "small", srcs = [ "custom_gradient_test.cc", ], args = ["--heap_check="], # TODO(b/174752220): Remove tags = tf_cuda_tests_tags(), deps = [ "//tensorflow/c:tf_status_helper",
Plain Text - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Mon Apr 01 20:39:44 GMT 2024 - 6.7K bytes - Viewed (0)