Searched defs:diag_index (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 31 static inline int ComputeDiagLen(int diag_index, int num_rows, int num_cols) { in ComputeDiagLen()
37 static inline bool IsLeftAligned(int diag_index, bool left_align_superdiagonal, in IsLeftAligned()
69 std::vector<int64_t> diag_index; in ProcessDiagIndex() local
164 for (int64_t diag_index = lower_diag_index; diag_index <= upper_diag_index; in SetMatrixDiag() local
411 for (int diag_index = upper_diag_index; diag_index >= lower_diag_index; in Compile() local
|
| /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
181 auto& diag_index = context->input(1); in Compute() local
341 int diag_index, int max_diag_len, int num_rows, int num_cols, in ComputeDiagLenAndContentOffset()
379 const int diag_index = j - i; in Compute() local
434 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
229 const Eigen::Index diag_index = upper_diag_index - m; in Compute() local
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseFactories.cpp | 71 Tensor diag_index = at::arange(n_diag, offsets_1d.options()); in spdiags() local
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | SparseFactories.cpp | 39 int64_t n_out) -> int64_t { in _spdiags_kernel_cpu()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 38 diag_index, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | diag_op_test.py | 42 diag_index, argument
|