Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
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()
91 TensorShapes input_matrix_shapes; in Compute() local
116 TensorShapes* input_matrix_shapes, in AnalyzeInputs()
160 OpKernelContext* context, const TensorShapes& input_matrix_shapes, in PrepareOutputs()
220 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.h59 OpKernelContext* context, const TensorShapes& input_matrix_shapes) const { in ValidateInputMatrixShapes()
90 const TensorShapes& input_matrix_shapes) const { in GetOutputMatrixShapes()
100 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()