Searched defs:indices_flat (Results 1 – 12 of 12) sorted by relevance
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
163 auto indices_flat = indices.flat<Index>(); in Compute() local
124 auto indices_flat = indices.flat<int32>(); in SparseSegmentMeanGradHelper() local
694 const auto indices_flat = op_indices->flat<Index>(); in Compute() local719 auto indices_flat = indices->flat<Index>(); in AddBatchOffsets() local904 auto indices_flat = indices.flat<Index>(); in DoCompute() local
118 auto indices_flat = indices.flat<Index>(); in DoCompute() local
188 GpuDeviceArrayOnHost<int32> indices_flat(c, first_dim_size); in Compute() local
912 auto indices_flat = index_flattener(c, indices); in DoScatterNd() local
3429 auto indices_flat = indices.flat<Tindex>(); in Compute() local
168 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in main() local
92 auto indices_flat = indices->template flat<int64>(); in Compute() local
290 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopDetections() local
1727 auto indices_flat = out_indices->flat<int64>(); in FastParseSingleExample() local