Home
last modified time | relevance | path

Searched defs:permutation_indices (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op_gpu.cu.cc41 Scalar* __restrict__ permutation_indices) { in ComputePermutationFromTranspositions()
136 Tensor* permutation_indices = nullptr; in ComputeAsync() local
Dlu_op.cc144 IndicesMap permutation_indices( in ComputeTensorSlice() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_ordering_amd_op.cc89 Tensor permutation_indices(cpu_allocator(), DT_INT32, in Compute() local
Dsparse_cholesky_op.cc231 const Tensor& permutation_indices, int* batch_size, in ValidateInputs()
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_ops_test.py63 def twist_matrix(matrix, permutation_indices): argument