Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlinalg_ops_common.cc37 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()
Dsvd_op_impl.h52 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()
Dmatrix_solve_ls_op_impl.h55 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()
Dlinalg_ops_common.h58 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()
Dqr_op_impl.h66 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()
Dself_adjoint_eig_v2_op_impl.h50 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()
Deig_op_impl.h55 const TensorShapes& input_matrix_shapes) const final { in GetOutputMatrixShapes()