| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 57 int64 upper_diag_index = 0; in ProcessDiagIndex() local
66 upper_diag_index = lower_diag_index; in ProcessDiagIndex()
77 upper_diag_index = in ProcessDiagIndex()
81 context, lower_diag_index <= upper_diag_index, in ProcessDiagIndex()
84 lower_diag_index, " > ", upper_diag_index)); in ProcessDiagIndex()
87 return {lower_diag_index, upper_diag_index}; in ProcessDiagIndex()
94 const int64 upper_diag_index, in ValidateDiagIndexWithOutputMatrixSize() argument
105 (-num_rows < upper_diag_index && upper_diag_index < num_cols) || in ValidateDiagIndexWithOutputMatrixSize()
106 upper_diag_index == 0, in ValidateDiagIndexWithOutputMatrixSize()
108 "upper_diag_index is out of bound: ", upper_diag_index, in ValidateDiagIndexWithOutputMatrixSize()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 64 int32 upper_diag_index = 0; in Compute() local
77 upper_diag_index = lower_diag_index; in Compute()
86 upper_diag_index = diag_index.flat<int32>()(1); in Compute()
111 (-num_rows < upper_diag_index && upper_diag_index < num_cols) || in Compute()
112 upper_diag_index == 0, in Compute()
114 "upper_diag_index is out of bound: ", upper_diag_index, in Compute()
117 context, lower_diag_index <= upper_diag_index, in Compute()
120 lower_diag_index, " > ", upper_diag_index)); in Compute()
126 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute()
129 std::min(num_rows + std::min(upper_diag_index, 0), in Compute()
[all …]
|
| D | matrix_set_diag_op.cc | 62 int32 upper_diag_index = 0; in Compute() local
74 upper_diag_index = lower_diag_index; in Compute()
83 upper_diag_index = diag_index.flat<int32>()(1); in Compute()
113 (-num_rows < upper_diag_index && upper_diag_index < num_cols) || in Compute()
114 upper_diag_index == 0, in Compute()
116 "upper_diag_index is out of bound: ", upper_diag_index, in Compute()
119 context, lower_diag_index <= upper_diag_index, in Compute()
122 lower_diag_index, " > ", upper_diag_index)); in Compute()
125 const Eigen::Index num_diags = upper_diag_index - lower_diag_index + 1; in Compute()
128 lower_diag_index == upper_diag_index || in Compute()
[all …]
|
| D | matrix_diag_op_gpu.cu.cc | 48 const int upper_diag_index, const T padding_value, in MatrixDiagKernel() argument
57 const int diag_index_in_input = upper_diag_index - diag_index; in MatrixDiagKernel()
62 if (lower_diag_index <= diag_index && diag_index <= upper_diag_index) { in MatrixDiagKernel()
78 const Eigen::Index upper_diag_index, in Compute()
85 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute()
95 num_diags, max_diag_len, lower_diag_index, upper_diag_index, in Compute()
105 const int upper_diag_index, const T padding_value, in MatrixDiagPartKernel() argument
115 const int diag_index = upper_diag_index - mapped_diag_index; in MatrixDiagPartKernel()
139 const Eigen::Index upper_diag_index, in Compute()
146 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute()
[all …]
|
| D | matrix_set_diag_op_gpu.cu.cc | 48 const int max_diag_len, const int upper_diag_index, in MatrixSetDiagKernel() argument
56 const int diag_index = upper_diag_index - diag_index_in_input; in MatrixSetDiagKernel()
75 const int upper_diag_index, const bool left_align_superdiagonal, in MatrixCopyInputAndSetDiagKernel() argument
84 const int diag_index_in_input = upper_diag_index - diag_index; in MatrixCopyInputAndSetDiagKernel()
89 if (lower_diag_index <= diag_index && diag_index <= upper_diag_index) { in MatrixCopyInputAndSetDiagKernel()
106 const Eigen::Index upper_diag_index, in Compute()
113 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute()
123 upper_diag_index, left_align_superdiagonal, left_align_subdiagonal, in Compute()
131 lower_diag_index, upper_diag_index, left_align_superdiagonal, in Compute()
|
| D | matrix_diag_op.h | 53 const Eigen::Index upper_diag_index, const Eigen::Index max_diag_len,
64 const Eigen::Index lower_diag_index, const Eigen::Index upper_diag_index,
|
| D | matrix_set_diag_op.h | 33 const Eigen::Index upper_diag_index,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1246 int32* lower_diag_index, int32* upper_diag_index) { in ReadDiagIndex() argument
1250 *upper_diag_index = *lower_diag_index; in ReadDiagIndex()
1255 *upper_diag_index = *lower_diag_index; in ReadDiagIndex()
1258 *upper_diag_index = diag_index_tensor->vec<int32>()(1); in ReadDiagIndex()
1281 int32 upper_diag_index = 0; in MatrixDiagPartV2Shape() local
1283 &upper_diag_index)); in MatrixDiagPartV2Shape()
1284 if (lower_diag_index > upper_diag_index) { in MatrixDiagPartV2Shape()
1300 if (upper_diag_index != 0 && // For when num_rows or num_cols == 0. in MatrixDiagPartV2Shape()
1301 (-num_rows >= upper_diag_index || upper_diag_index >= num_cols)) { in MatrixDiagPartV2Shape()
1304 max_diag_len = std::min(num_rows + std::min(upper_diag_index, 0), in MatrixDiagPartV2Shape()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 746 lower_diag_index, upper_diag_index = diags
747 num_diags = upper_diag_index - lower_diag_index + 1
|