Commit aadc9ce4 authored by Adam Procter's avatar Adam Procter Committed by Robert Kimball

Fix element type for create_tensor of cached fprop nodes in backprop_derivative (#862)

parent d60cd1d5
......@@ -1015,7 +1015,7 @@ TEST(${BACKEND_NAME}, backwards_reshape)
EXPECT_TRUE(autodiff_numeric_compare<float>(backend, make_graph, {x0}, .01f, .01f));
}
TEST(DISABLED_${BACKEND_NAME}, backwards_select)
TEST(${BACKEND_NAME}, backwards_select)
{
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("NNP_TESTER", "${BACKEND_NAME}");
......@@ -1044,7 +1044,7 @@ TEST(DISABLED_${BACKEND_NAME}, backwards_select)
}
}
TEST(DISABLED_${BACKEND_NAME}, backwards_select_nested)
TEST(${BACKEND_NAME}, backwards_select_nested)
{
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("NNP_TESTER", "${BACKEND_NAME}");
......
......@@ -179,7 +179,7 @@ namespace ngraph
// add cached nodes to both modified f output and modified f' input arguments
for (auto node : fprop_cache.fprop_output_nodes)
{
auto tv = backend->create_tensor<T>(node->get_shape());
auto tv = backend->create_tensor(node->get_element_type(), node->get_shape());
mod_f_output_args.push_back(tv);
mod_df_input_args.push_back(tv);
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment