Home
last modified time | relevance | path

Searched refs:Tindices (Results 1 – 25 of 172) sorted by relevance

1234567

/external/tensorflow/tensorflow/core/kernels/
Dsparse_utils.cc25 template <typename Tindices>
26 Tindices FindNextDenseRowStartIndex( in FindNextDenseRowStartIndex()
27 const Tindices sparse_index_begin, in FindNextDenseRowStartIndex()
28 const typename TTypes<Tindices>::ConstMatrix& indices_mat) { in FindNextDenseRowStartIndex()
30 Tindices begin = sparse_index_begin; in FindNextDenseRowStartIndex()
31 Tindices end = indices_mat.dimension(0); in FindNextDenseRowStartIndex()
32 const Tindices orig_sparse_index_end = end; in FindNextDenseRowStartIndex()
35 const Tindices orig_dense_index_begin = indices_mat(begin, 0); in FindNextDenseRowStartIndex()
41 Tindices increment = 1; in FindNextDenseRowStartIndex()
54 const Tindices dense_row_index_to_find = orig_dense_index_begin; in FindNextDenseRowStartIndex()
[all …]
Dsparse_utils.h37 template <typename Tindices>
38 Tindices FindNextDenseRowStartIndex(
39 const Tindices sparse_index_begin,
40 const typename TTypes<Tindices>::ConstMatrix& indices_mat);
48 template <typename Tindices>
49 std::vector<Tindices> GetStartIndicesOfEachDenseRow(
50 const typename TTypes<Tindices>::ConstMatrix& indices_mat,
55 template <typename Tindices>
56 std::vector<Tindices> ParseRowStartIndices(
58 const Tindices num_nonzero_entries_in_sparse_mat);
[all …]
Dtensor_flag_utils.h32 template <typename Tindices>
33 std::vector<Tindices> ParseRowStartIndices(
35 const Tindices num_nonzero_entries_in_sparse_mat);
67 template <typename Tindices>
68 Tindices GetLinearBucket(const Tindices value, const Tindices bucket_size);
72 template <typename Tindices>
73 Tindices GetPowerBucket(const Tindices value, const Tindices bucket_size);
Dsparse_tensor_dense_matmul_op_gpu.cu.cc29 template <typename T, typename Tindices, bool ADJ_A, bool ADJ_B>
32 const Tindices* __restrict__ a_indices, const T* __restrict__ a_values, in SparseTensorDenseMatMulKernel()
63 template <typename T, typename Tindices, bool ADJ_A, bool ADJ_B>
64 struct SparseTensorDenseMatMulFunctor<GPUDevice, T, Tindices, ADJ_A, ADJ_B> {
67 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute()
83 SparseTensorDenseMatMulKernel<T, Tindices, ADJ_A, ADJ_B>, in Compute()
94 #define DEFINE(T, Tindices) \ argument
96 GPUDevice, T, Tindices, false, false>; \
98 GPUDevice, T, Tindices, false, true>; \
100 GPUDevice, T, Tindices, true, false>; \
[all …]
Dtensor_flag_utils.cc144 template <typename Tindices>
145 Tindices GetLinearBucket(const Tindices value, const Tindices bucket_size) { in GetLinearBucket()
146 const Tindices next_multiple_of_bucket_size = in GetLinearBucket()
151 template <typename Tindices>
152 Tindices GetPowerBucket(const Tindices value, const Tindices bucket_size) { in GetPowerBucket()
Dsparse_tensor_dense_matmul_op.cc33 template <typename Device, typename T, typename Tindices>
139 Device, T, Tindices, ADJ_A, \ in Compute()
141 a_indices->matrix<Tindices>(), a_values->vec<T>(), \ in Compute()
182 #define DECLARE_GPU_SPEC(T, Tindices, ADJ_A, ADJ_B) \ argument
185 GPUDevice, T, Tindices, ADJ_A, \
187 TTypes<Tindices>::ConstMatrix a_indices, \
191 GPUDevice, T, Tindices, ADJ_A, ADJ_B>;
244 template <typename T, typename Tindices, bool ADJ_A, bool ADJ_B>
245 struct SparseTensorDenseMatMulFunctor<CPUDevice, T, Tindices, ADJ_A, ADJ_B> {
250 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute()
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSparseTensorDenseAdd.pbtxt5 type_attr: "Tindices"
13 type_attr: "Tindices"
46 name: "Tindices"
60 type_attr: "Tindices"
68 type_attr: "Tindices"
103 name: "Tindices"
117 type_attr: "Tindices"
125 type_attr: "Tindices"
161 name: "Tindices"
175 type_attr: "Tindices"
[all …]
DSegmentMean.pbtxt9 type_attr: "Tindices"
33 name: "Tindices"
51 type_attr: "Tindices"
77 name: "Tindices"
95 type_attr: "Tindices"
122 name: "Tindices"
140 type_attr: "Tindices"
167 name: "Tindices"
185 type_attr: "Tindices"
217 name: "Tindices"
DScatterNdNonAliasingAdd.pbtxt9 type_attr: "Tindices"
42 name: "Tindices"
60 type_attr: "Tindices"
95 name: "Tindices"
113 type_attr: "Tindices"
149 name: "Tindices"
167 type_attr: "Tindices"
203 name: "Tindices"
221 type_attr: "Tindices"
258 name: "Tindices"
DSegmentMin.pbtxt9 type_attr: "Tindices"
33 name: "Tindices"
51 type_attr: "Tindices"
77 name: "Tindices"
95 type_attr: "Tindices"
122 name: "Tindices"
140 type_attr: "Tindices"
167 name: "Tindices"
DSegmentMax.pbtxt9 type_attr: "Tindices"
33 name: "Tindices"
51 type_attr: "Tindices"
77 name: "Tindices"
95 type_attr: "Tindices"
122 name: "Tindices"
140 type_attr: "Tindices"
167 name: "Tindices"
DResourceScatterUpdate.pbtxt9 type_attr: "Tindices"
40 name: "Tindices"
59 type_attr: "Tindices"
91 name: "Tindices"
110 type_attr: "Tindices"
142 name: "Tindices"
161 type_attr: "Tindices"
172 name: "Tindices"
DSegmentProd.pbtxt9 type_attr: "Tindices"
38 name: "Tindices"
56 type_attr: "Tindices"
87 name: "Tindices"
105 type_attr: "Tindices"
137 name: "Tindices"
155 type_attr: "Tindices"
187 name: "Tindices"
DSegmentSum.pbtxt9 type_attr: "Tindices"
38 name: "Tindices"
56 type_attr: "Tindices"
87 name: "Tindices"
105 type_attr: "Tindices"
137 name: "Tindices"
155 type_attr: "Tindices"
187 name: "Tindices"
DUnsortedSegmentMax.pbtxt9 type_attr: "Tindices"
37 name: "Tindices"
55 type_attr: "Tindices"
85 name: "Tindices"
103 type_attr: "Tindices"
134 name: "Tindices"
165 type_attr: "Tindices"
196 name: "Tindices"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSparseTensorDenseAdd.pbtxt5 type_attr: "Tindices"
13 type_attr: "Tindices"
46 name: "Tindices"
60 type_attr: "Tindices"
68 type_attr: "Tindices"
103 name: "Tindices"
117 type_attr: "Tindices"
125 type_attr: "Tindices"
161 name: "Tindices"
175 type_attr: "Tindices"
[all …]
DSegmentMean.pbtxt9 type_attr: "Tindices"
33 name: "Tindices"
51 type_attr: "Tindices"
77 name: "Tindices"
95 type_attr: "Tindices"
122 name: "Tindices"
140 type_attr: "Tindices"
167 name: "Tindices"
185 type_attr: "Tindices"
217 name: "Tindices"
DScatterNdNonAliasingAdd.pbtxt9 type_attr: "Tindices"
42 name: "Tindices"
60 type_attr: "Tindices"
95 name: "Tindices"
113 type_attr: "Tindices"
149 name: "Tindices"
167 type_attr: "Tindices"
203 name: "Tindices"
221 type_attr: "Tindices"
258 name: "Tindices"
DSegmentMax.pbtxt9 type_attr: "Tindices"
33 name: "Tindices"
51 type_attr: "Tindices"
77 name: "Tindices"
95 type_attr: "Tindices"
122 name: "Tindices"
140 type_attr: "Tindices"
167 name: "Tindices"
DSegmentMin.pbtxt9 type_attr: "Tindices"
33 name: "Tindices"
51 type_attr: "Tindices"
77 name: "Tindices"
95 type_attr: "Tindices"
122 name: "Tindices"
140 type_attr: "Tindices"
167 name: "Tindices"
DResourceScatterUpdate.pbtxt9 type_attr: "Tindices"
40 name: "Tindices"
59 type_attr: "Tindices"
91 name: "Tindices"
110 type_attr: "Tindices"
142 name: "Tindices"
161 type_attr: "Tindices"
172 name: "Tindices"
DSegmentProd.pbtxt9 type_attr: "Tindices"
38 name: "Tindices"
56 type_attr: "Tindices"
87 name: "Tindices"
105 type_attr: "Tindices"
137 name: "Tindices"
155 type_attr: "Tindices"
187 name: "Tindices"
DSegmentSum.pbtxt9 type_attr: "Tindices"
38 name: "Tindices"
56 type_attr: "Tindices"
87 name: "Tindices"
105 type_attr: "Tindices"
137 name: "Tindices"
155 type_attr: "Tindices"
187 name: "Tindices"
DResourceScatterAdd.pbtxt9 type_attr: "Tindices"
38 name: "Tindices"
57 type_attr: "Tindices"
88 name: "Tindices"
107 type_attr: "Tindices"
139 name: "Tindices"
158 type_attr: "Tindices"
190 name: "Tindices"
DUnsortedSegmentMax.pbtxt9 type_attr: "Tindices"
37 name: "Tindices"
55 type_attr: "Tindices"
85 name: "Tindices"
103 type_attr: "Tindices"
134 name: "Tindices"
165 type_attr: "Tindices"
196 name: "Tindices"

1234567