Home
last modified time | relevance | path

Searched defs:num_eqs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_matmul_op.cc68 const int num_eqs = static_cast<int>(input_matrix_shapes[0].dim_size(1)); in GetCostPerUnit() local
Dtridiagonal_solve_op.cc88 const int num_eqs = static_cast<int>(input_matrix_shapes[0].dim_size(1)); in GetCostPerUnit() local
Dtridiagonal_solve_op_gpu.cu.cc207 const Scalar* subdiag, Scalar* rhs, const int num_eqs, in SolveWithGtsv()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal_test.cc114 const int64_t num_eqs = std::get<1>(spec); in XLA_TEST_P() local