Searched defs:input_indices_t (Results 1 – 2 of 2) sorted by relevance
511 auto input_indices_t = input_tensor.indices().matrix<int64_t>(); in Split() local608 auto input_indices_t = input_tensor.indices().matrix<int64_t>(); in Slice() local
456 auto input_indices_t = To32Bit(input_indices.flat<int32>()); in LaunchSortKernel() local