Searched defs:max_diag_len (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_set_diag_op_gpu.cu.cc | 31 const int max_diag_len, in ComputeContentOffset()
48 const int max_diag_len, const int upper_diag_index, in MatrixSetDiagKernel()
74 const int max_diag_len, const int lower_diag_index, in MatrixCopyInputAndSetDiagKernel()
|
| D | matrix_diag_op_gpu.cu.cc | 30 const int max_diag_len, in ComputeContentOffset()
47 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagKernel()
104 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagPartKernel()
|
| D | matrix_diag_op.cc | 134 const int32_t max_diag_len = in Compute() local
242 const Eigen::Index max_diag_len = diagonal_shape.dim_size(diag_rank - 1); in Compute() local
341 int diag_index, int max_diag_len, int num_rows, int num_cols, in ComputeDiagLenAndContentOffset()
|
| D | matrix_set_diag_op.cc | 140 const int32_t max_diag_len = in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 123 const int64_t max_diag_len, const int64_t num_rows, in SetMatrixDiag()
277 const int64_t max_diag_len = diag_shape.dim_size(diag_rank - 1); in Compile() local
395 const int32_t max_diag_len = in Compile() local
510 const int32_t max_diag_len = in Compile() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1384 int32_t max_diag_len = InferenceContext::kUnknownDim; in MatrixDiagPartV2Shape() local
1468 const int32_t max_diag_len = c->Value(c->Dim(input_shape, input_rank - 1)); in MatrixDiagV2Shape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1921 int64_t max_diag_len = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3350 int64_t max_diag_len = std::min(shape[rank - 2] + std::min(k1, 0), in TEST_F() local
|