Searched defs:max_diag_len (Results 1 – 7 of 7) 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 | 128 const int32 max_diag_len = in Compute() local
220 const Eigen::Index max_diag_len = diagonal_shape.dim_size(diag_rank - 1); in Compute() local
319 int diag_index, int max_diag_len, int num_rows, int num_cols, in ComputeDiagLenAndContentOffset()
|
| D | matrix_set_diag_op.cc | 137 const int32 max_diag_len = in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 120 const int64 upper_diag_index, const int64 max_diag_len, in SetMatrixDiag()
274 const int64 max_diag_len = diag_shape.dim_size(diag_rank - 1); in Compile() local
392 const int32 max_diag_len = in Compile() local
507 const int32 max_diag_len = in Compile() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1293 int32 max_diag_len = InferenceContext::kUnknownDim; in MatrixDiagPartV2Shape() local
1377 const int32 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 | 1520 int64_t max_diag_len = in matchAndRewrite() local
|