Home
last modified time | relevance | path

Searched refs:y_shapes (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc107 const std::vector<TensorShape>& y_shapes, in ComputeTheoreticalJacobianTranspose() argument
109 size_t y_num = y_shapes.size(); in ComputeTheoreticalJacobianTranspose()
113 dys.reserve(y_shapes.size()); in ComputeTheoreticalJacobianTranspose()
114 for (const auto& y_shape : y_shapes) { in ComputeTheoreticalJacobianTranspose()
125 dy_datas[i] = Tensor(ys[i].type(), y_shapes[i]); in ComputeTheoreticalJacobianTranspose()
147 const int64 dy_size = y_shapes[y_idx].num_elements(); in ComputeTheoreticalJacobianTranspose()
208 const std::vector<TensorShape>& y_shapes, in ComputeNumericJacobianTranspose() argument
212 size_t y_num = y_shapes.size(); in ComputeNumericJacobianTranspose()
249 const int64 y_size = y_shapes[y_idx].num_elements(); in ComputeNumericJacobianTranspose()
298 const std::vector<TensorShape>& y_shapes, in InitJacobians() argument
[all …]
Dgradient_checker.h52 const std::vector<TensorShape>& y_shapes,
/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad_test.cc38 const OutputList& ys, const std::vector<TensorShape>& y_shapes) { in RunTest() argument
42 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()
Darray_grad_test.cc45 const OutputList& ys, const std::vector<TensorShape>& y_shapes) { in RunTest() argument
49 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()
80 std::vector<TensorShape> y_shapes(4, TensorShape({2, 3})); in TEST_F() local
82 RunTest({x}, {x_shape}, y.output, y_shapes); in TEST_F()
89 std::vector<TensorShape> y_shapes(2, TensorShape({4, 3})); in TEST_F() local
91 RunTest({x}, {x_shape}, y.output, y_shapes); in TEST_F()
Dnn_grad_test.cc71 const OutputList& ys, const std::vector<TensorShape>& y_shapes) { in RunTest() argument
75 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()
Dmath_grad_test.cc714 const OutputList& ys, const std::vector<TensorShape>& y_shapes) { in RunTest() argument
718 scope_, xs, x_shapes, ys, y_shapes, &max_error))); in RunTest()