| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 56 int64 lower_diag_index = 0; in ProcessDiagIndex() local
65 context->ConstantInputAsIntScalar("k", &lower_diag_index)); in ProcessDiagIndex()
66 upper_diag_index = lower_diag_index; in ProcessDiagIndex()
76 lower_diag_index = diag_index[0]; in ProcessDiagIndex()
78 (diag_index.size() > 1) ? diag_index[1] : lower_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()
93 const int64 lower_diag_index, in ValidateDiagIndexWithOutputMatrixSize() argument
99 (-num_rows < lower_diag_index && lower_diag_index < num_cols) || in ValidateDiagIndexWithOutputMatrixSize()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 63 int32 lower_diag_index = 0; in Compute() local
76 lower_diag_index = diag_index.flat<int32>()(0); in Compute()
77 upper_diag_index = lower_diag_index; in Compute()
105 (-num_rows < lower_diag_index && lower_diag_index < num_cols) || in Compute()
106 lower_diag_index == 0, in Compute()
108 "lower_diag_index is out of bound: ", lower_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()
130 num_cols - std::max(lower_diag_index, 0)); in Compute()
[all …]
|
| D | matrix_set_diag_op.cc | 61 int32 lower_diag_index = 0; in Compute() local
73 lower_diag_index = diag_index.flat<int32>()(0); in Compute()
74 upper_diag_index = lower_diag_index; in Compute()
107 (-num_rows < lower_diag_index && lower_diag_index < num_cols) || in Compute()
108 lower_diag_index == 0, in Compute()
110 "lower_diag_index is out of bound: ", lower_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 | 47 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagKernel() argument
62 if (lower_diag_index <= diag_index && diag_index <= upper_diag_index) { in MatrixDiagKernel()
77 const Eigen::Index lower_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()
104 const int num_diags, const int max_diag_len, const int lower_diag_index, in MatrixDiagPartKernel() argument
138 const Eigen::Index lower_diag_index, in Compute()
146 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute()
156 num_diags, max_diag_len, lower_diag_index, upper_diag_index, in Compute()
|
| D | matrix_set_diag_op_gpu.cu.cc | 74 const int max_diag_len, const int lower_diag_index, in MatrixCopyInputAndSetDiagKernel() argument
89 if (lower_diag_index <= diag_index && diag_index <= upper_diag_index) { in MatrixCopyInputAndSetDiagKernel()
105 const Eigen::Index lower_diag_index, in Compute()
113 const int num_diags = upper_diag_index - lower_diag_index + 1; in Compute()
131 lower_diag_index, upper_diag_index, left_align_superdiagonal, in Compute()
|
| D | matrix_diag_op.h | 52 typename TTypes<T>::Tensor& output, const Eigen::Index lower_diag_index,
64 const Eigen::Index lower_diag_index, const Eigen::Index upper_diag_index,
|
| D | matrix_set_diag_op.h | 32 const Eigen::Index lower_diag_index,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1246 int32* lower_diag_index, int32* upper_diag_index) { in ReadDiagIndex() argument
1249 *lower_diag_index = diag_index_tensor->scalar<int32>()(); in ReadDiagIndex()
1250 *upper_diag_index = *lower_diag_index; in ReadDiagIndex()
1254 *lower_diag_index = diag_index_tensor->vec<int32>()(0); in ReadDiagIndex()
1255 *upper_diag_index = *lower_diag_index; in ReadDiagIndex()
1257 *lower_diag_index = diag_index_tensor->vec<int32>()(0); in ReadDiagIndex()
1280 int32 lower_diag_index = 0; in MatrixDiagPartV2Shape() local
1282 TF_RETURN_IF_ERROR(ReadDiagIndex(c, diag_index_tensor, &lower_diag_index, in MatrixDiagPartV2Shape()
1284 if (lower_diag_index > upper_diag_index) { in MatrixDiagPartV2Shape()
1296 if (lower_diag_index != 0 && // For when num_rows or num_cols == 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
|