Searched refs:permutation_indices (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | lu_op_gpu.cu.cc | 41 Scalar* __restrict__ permutation_indices) { in ComputePermutationFromTranspositions() argument
44 permutation_indices[i] = Scalar(i); in ComputePermutationFromTranspositions()
54 Scalar t = permutation_indices[i]; in ComputePermutationFromTranspositions()
55 permutation_indices[i] = permutation_indices[pivots[i] - 1]; in ComputePermutationFromTranspositions()
56 permutation_indices[pivots[i] - 1] = t; in ComputePermutationFromTranspositions()
136 Tensor* permutation_indices = nullptr; in ComputeAsync() local
139 &permutation_indices), in ComputeAsync()
226 permutation_indices->template flat<Tidx>().data(); in ComputeAsync()
|
| D | lu_op.cc | 144 IndicesMap permutation_indices( in ComputeTensorSlice() local
162 permutation_indices = permutation.indices(); in ComputeTensorSlice()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_ordering_amd_op.cc | 89 Tensor permutation_indices(cpu_allocator(), DT_INT32, in Compute() local
91 ctx->set_output(0, permutation_indices); in Compute()
120 permutation_indices.flat<int>().data() + batch_index * num_rows, in Compute()
|
| D | sparse_cholesky_op.cc | 231 const Tensor& permutation_indices, int* batch_size, in ValidateInputs() argument
251 const TensorShape& perm_shape = permutation_indices.shape(); in ValidateInputs()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 63 def twist_matrix(matrix, permutation_indices): argument
69 permutation_indices_inv = array_ops.invert_permutation(permutation_indices)
74 permutation_indices)
79 math_ops.range(batch_size)[:, None], permutation_indices.shape)
|