Searched defs:transposed_shape (Results 1 – 10 of 10) sorted by relevance
178 TensorShape transposed_shape = input.shape(); in ComputeAsync() local
223 TensorShape transposed_shape; in TransposeOperand() local
163 TensorShape transposed_shape(reshaped_shape[1], reshaped_shape[0]); in validate() local
214 std::vector<int64_t> transposed_shape = merged_shape; in Subgroup() local
395 RuntimeShape transposed_shape(GetTensorShape(tensor_in)); in TransposeRowsColumnsImpl() local
90 SmallVector<int64_t, 4> transposed_shape(shape.begin(), shape.end()); in createTransposeOp() local
112 llvm::SmallVector<int64_t> transposed_shape(permutation_array.size()); in GetPermutationAndTransposedShape() local523 llvm::SmallVector<int64_t, 4> transposed_shape(num_spatial_dims + 2); in GetReformatTypeAndPermutation() local
467 TensorShape transposed_shape = input.shape(); in TransposeAndConjugateTensor() local
3370 llvm::SmallVector<int64_t, 4> transposed_shape; in verify() local
1447 llvm::SmallVector<int64_t, 2> transposed_shape = {2, input_rank}; in matchAndRewrite() local