Home
last modified time | relevance | path

Searched refs:GetCostPerUnit (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op.cc58 int64 GetCostPerUnit(const TensorShape& input_matrix_shape) const { in GetCostPerUnit() function in tensorflow::LuOp
124 batch_shape.num_elements(), GetCostPerUnit(input_matrix_shape), in Compute()
Dtridiagonal_matmul_op.cc67 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function in tensorflow::TridiagonalMatMulOp
Dsvd_op_impl.h71 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function
Dmatrix_solve_ls_op_impl.h65 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function
Dlinalg_ops_common.h99 virtual int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const { in GetCostPerUnit() function
Dtridiagonal_solve_op.cc77 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function in tensorflow::TridiagonalSolveOp
Dlinalg_ops_common.cc113 batch_shape.num_elements(), GetCostPerUnit(input_matrix_shapes), shard); in Compute()
Dqr_op_impl.h83 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function
Dmatrix_solve_op.cc65 int64 GetCostPerUnit(const TensorShapes& input_matrix_shapes) const final { in GetCostPerUnit() function in tensorflow::MatrixSolveOp