Searched defs:indices_flat (Results 1 – 12 of 12) sorted by relevance
107 auto indices_flat = indices.flat<Index>(); in Compute() local
124 auto indices_flat = indices.flat<Index>(); in DoCompute() local212 auto indices_flat = indices_host.flat<Index>(); in DoCompute() local
124 auto indices_flat = indices.flat<int32>(); in SparseSegmentMeanGradHelper() local
600 auto indices_flat = indices->flat<Index>(); in AddBatchOffsets() local690 const auto indices_flat = op_indices->flat<Index>(); in Compute() local785 auto indices_flat = indices.flat<Index>(); in Compute() local
188 GpuDeviceArrayOnHost<int32> indices_flat(c, first_dim_size); in Compute() local
672 auto indices_flat = index_flattener(c, indices); in DoScatterNd() local
247 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local265 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in CheckTopLabel() local
167 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in main() local
272 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local290 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in CheckTopLabel() local
286 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopDetections() local
316 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local
1502 auto indices_flat = out_indices->flat<int64>(); in FastParseSingleExample() local