Searched defs:diag_index (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_set_diag_op_gpu.cu.cc | 30 __device__ inline int ComputeContentOffset(const int diag_index, in ComputeContentOffset()
56 const int diag_index = upper_diag_index - diag_index_in_input; in MatrixSetDiagKernel() local
83 const int diag_index = col - row; in MatrixCopyInputAndSetDiagKernel() local
|
| D | matrix_diag_op.cc | 69 auto& diag_index = context->input(1); in Compute() local
175 auto& diag_index = context->input(1); in Compute() local
319 int diag_index, int max_diag_len, int num_rows, int num_cols, in ComputeDiagLenAndContentOffset()
357 const int diag_index = j - i; in Compute() local
412 const Eigen::Index diag_index = upper_diag_index - m; in Compute() local
|
| D | matrix_diag_op_gpu.cu.cc | 29 __device__ inline int ComputeContentOffset(const int diag_index, in ComputeContentOffset()
56 const int diag_index = col - row; in MatrixDiagKernel() local
115 const int diag_index = upper_diag_index - mapped_diag_index; in MatrixDiagPartKernel() local
|
| D | matrix_set_diag_op.cc | 66 auto& diag_index = context->input(2); in Compute() local
226 const Eigen::Index diag_index = upper_diag_index - m; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 30 static inline int ComputeDiagLen(int diag_index, int num_rows, int num_cols) { in ComputeDiagLen()
36 static inline bool IsLeftAligned(int diag_index, bool left_align_superdiagonal, in IsLeftAligned()
68 std::vector<int64> diag_index; in ProcessDiagIndex() local
161 for (int64 diag_index = lower_diag_index; diag_index <= upper_diag_index; in SetMatrixDiag() local
408 for (int diag_index = upper_diag_index; diag_index >= lower_diag_index; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 41 diag_index, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 46 diag_index, argument
|