Searched refs:permutation_indices_shape (Results 1 – 2 of 2) sorted by relevance
87 TensorShape permutation_indices_shape = in Compute()90 permutation_indices_shape); in Compute()
107 TensorShape permutation_indices_shape = batch_shape; in ComputeAsync() local108 permutation_indices_shape.AddDim(num_rows); in ComputeAsync()138 context->allocate_output(1, permutation_indices_shape, in ComputeAsync()