Home
last modified time | relevance | path

Searched defs:GetCostPerUnit (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dsvd_op_impl.h71 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function
H A Dmatrix_solve_ls_op_impl.h65 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function
H A Dtridiagonal_matmul_op.cc67 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function in tensorflow::TridiagonalMatMulOp
H A Dlinalg_ops_common.h99 virtual int64_t GetCostPerUnit( in GetCostPerUnit() function
H A Dlu_op.cc58 int64_t GetCostPerUnit(const TensorShape& input_matrix_shape) const { in GetCostPerUnit() function in tensorflow::LuOp
H A Dmatrix_solve_op.cc65 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function in tensorflow::MatrixSolveOp
H A Dqr_op_impl.h83 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function
H A Dtridiagonal_solve_op.cc87 int64_t GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function in tensorflow::TridiagonalSolveOp