Home
last modified time | relevance | path

Searched defs:x_shapes (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dgradient_checker.cc106 const std::vector<TensorShape>& x_shapes, in ComputeTheoreticalJacobianTranspose()
213 const std::vector<TensorShape>& x_shapes, in ComputeNumericJacobianTranspose()
304 const std::vector<TensorShape>& x_shapes, in InitJacobians()
333 const std::vector<TensorShape>& x_shapes, in ComputeGradientErrorInternal()
379 const std::vector<TensorShape>& x_shapes, in ComputeGradientError()
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dfunctional_grad_test.cc47 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest()
H A Ddata_flow_grad_test.cc37 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest()
H A Dlinalg_grad_test.cc44 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest()
H A Dnn_grad_test.cc75 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest()
H A Darray_grad_test.cc44 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest()
H A Dmath_grad_test.cc744 void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, in RunTest()