Searched defs:indices_flat (Results 1 – 12 of 12) sorted by relevance
759 const auto indices_flat = op_indices->flat<Index>(); in Compute() local790 auto indices_flat = indices->flat<Index>(); in AddBatchOffsets() local1013 auto indices_flat = indices.flat<Index>(); in DoScatter() local1104 auto indices_flat = indices.flat<Index>(); in DoCompute() local
163 auto indices_flat = indices.flat<Index>(); in Compute() local
128 auto indices_flat = indices.flat<Index>(); in DoCompute() local
194 auto indices_flat = indices.flat<int32>(); in SparseSegmentMeanGradHelper() local
189 GpuDeviceArrayOnHost<int32> indices_flat(c, first_dim_size); in Compute() local
961 auto indices_flat = index_flattener(c, indices); in DoScatterNd() local
3445 auto indices_flat = indices.flat<Tindex>(); in Compute() local
248 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local266 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in CheckTopLabel() local
168 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in main() local
92 auto indices_flat = indices->template flat<int64_t>(); in Compute() local
290 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopDetections() local
1743 auto indices_flat = out_indices->flat<int64_t>(); in FastParseSingleExample() local