Home
last modified time | relevance | path

Searched refs:num_rhss (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmatrix_triangular_solve_op.cc70 double num_rhss = static_cast<double>(input_matrix_shapes[1].dim_size(1)); in GetCostPerUnit() local
71 double cost = rows * rows * num_rhss * in GetCostPerUnit()
162 double num_rhss = static_cast<double>(input_matrix_shapes[1].dim_size(1)); in GetCostPerUnit() local
163 double cost = rows * rows * num_rhss * in GetCostPerUnit()
Dtridiagonal_solve_op.cc68 const int num_rhss = static_cast<int>(input_matrix_shapes[1].dim_size(0)); in GetCostPerUnit() local
76 num_eqs * (div_cost * (num_rhss + 1) + in GetCostPerUnit()
77 (add_cost + mult_cost) * (2.5 * num_rhss + 1.5)); in GetCostPerUnit()
Dmatrix_solve_ls_op_impl.h68 double num_rhss = static_cast<double>(input_matrix_shapes[1].dim_size(1)); in GetCostPerUnit() local
69 double cost = std::max(m, n) * std::min(m, n) * (std::min(m, n) + num_rhss); in GetCostPerUnit()
Dmatrix_solve_op.cc67 double num_rhss = static_cast<double>(input_matrix_shapes[1].dim_size(1)); in GetCostPerUnit() local
68 double cost = rows * rows * (rows + num_rhss); in GetCostPerUnit()