Searched refs:permutation_shape (Results 1 – 2 of 2) sorted by relevance
93 TensorShape permutation_shape = batch_shape; in Compute() local94 permutation_shape.AddDim(num_rows); in Compute()96 TensorShapes output_matrix_shapes({input.shape(), permutation_shape}); in Compute()106 OP_REQUIRES_OK(context, context->allocate_output(1, permutation_shape, in Compute()
234 const int permutation_shape[] = {1, 1, 1, 4}; in PopulateSubGraph() local236 permutation_shape, reinterpret_cast<char*>(permutation), in PopulateSubGraph()