Searched refs:unfused_tensors (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 231 std::vector<Tensor> unfused_tensors; in CompareFusedAndSeparate() local
232 TF_ASSERT_OK(session->Run({}, {"conv"}, {}, &unfused_tensors)); in CompareFusedAndSeparate()
237 test::ExpectClose(unfused_tensors[0], fused_tensors[0]); in CompareFusedAndSeparate()
284 std::vector<Tensor> unfused_tensors; in CompareFusedPadOnlyAndSeparate() local
285 TF_ASSERT_OK(session->Run({}, {"conv"}, {}, &unfused_tensors)); in CompareFusedPadOnlyAndSeparate()
290 test::ExpectClose(unfused_tensors[0], fused_tensors[0]); in CompareFusedPadOnlyAndSeparate()
603 std::vector<Tensor> unfused_tensors; in RunAndFetch() local
604 TF_ASSERT_OK(session->Run({}, {fetch}, {}, &unfused_tensors)); in RunAndFetch()
606 *output = unfused_tensors[0]; in RunAndFetch()
|
| D | mkl_fused_ops_test.cc | 77 std::vector<Tensor> unfused_tensors; in RunAndFetch() local
78 TF_ASSERT_OK(session->Run({}, {fetch}, {}, &unfused_tensors)); in RunAndFetch()
80 *output = unfused_tensors[0]; in RunAndFetch()
|