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.cc34 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSingleMatrix()
45 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSingleSquareMatrix()
56 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSolver()
73 OpKernelContext* context, const TensorShapes& input_matrix_shapes) { in ValidateSquareSolver()
92 TensorShapes input_matrix_shapes; in Compute() local
119 TensorShapes* input_matrix_shapes, TensorShape* batch_shape) { in AnalyzeInputs()
162 OpKernelContext* context, const TensorShapes& input_matrix_shapes, in PrepareOutputs()
222 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 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 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()
99 virtual int64 GetCostPerUnit(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 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()