Searched defs:num_diags (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op_gpu.cu.cc | 47 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagKernel()
85 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
104 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagPartKernel()
146 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | matrix_set_diag_op_gpu.cu.cc | 47 const int num_threads, const int m, const int n, const int num_diags, in MatrixSetDiagKernel()
73 const int num_threads, const int m, const int n, const int num_diags, in MatrixCopyInputAndSetDiagKernel()
113 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | matrix_diag_op.cc | 132 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
225 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
373 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
419 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | matrix_set_diag_op.cc | 128 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
219 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute() local
|
| D | tridiagonal_solve_op.cc | 66 auto num_diags = input_matrix_shapes[0].dim_size(0); in ValidateInputMatrixShapes() local
|
| D | tridiagonal_solve_op_gpu.cu.cc | 119 auto num_diags = input_matrix_shapes[0].dim_size(0); in ValidateInputMatrixShapes() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 120 const int64_t diag_rank, const int64_t num_diags, in SetMatrixDiag()
278 const int64_t num_diags = upper_diag_index - lower_diag_index + 1; in Compile() local
393 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compile() local
498 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compile() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1441 const int32_t num_diags = c->Value(c->Dim(input_shape, input_rank - 2)); in MatrixDiagV2Shape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1916 int num_diags = k[1] - k[0] + 1; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3344 int64_t num_diags = in TEST_F() local
|