Searched defs:input_matrix_shapes (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | linalg_ops_common.cc | 37 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSingleMatrix()
48 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSingleSquareMatrix()
59 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSolver()
76 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSquareSolver()
95 TensorShapes input_matrix_shapes; in Compute() local
122 TensorShapes* input_matrix_shapes, TensorShape* batch_shape) { in AnalyzeInputs()
169 OpKernelContext* context, const TensorShapes& input_matrix_shapes, in PrepareOutputs()
234 const TensorShapes& input_matrix_shapes, const TensorOutputs& outputs, in ComputeTensorSlice()
|
| H A D | svd_op_impl.h | 52 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
57 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
71 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| H A D | matrix_solve_ls_op_impl.h | 55 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
60 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
65 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| H A D | linalg_ops_common.h | 58 OpKernelContext* context, const TensorShapes& input_matrix_shapes) const { in ValidateInputMatrixShapes()
89 const TensorShapes& input_matrix_shapes) const { in GetOutputMatrixShapes()
100 const TensorShapes& input_matrix_shapes) const { in GetCostPerUnit()
|
| H A D | qr_op_impl.h | 66 const TensorShapes& input_matrix_shapes) const final { in ValidateInputMatrixShapes()
71 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
83 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit()
|
| H A D | self_adjoint_eig_v2_op_impl.h | 50 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
|
| H A D | eig_op_impl.h | 55 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
|