Home
last modified time | relevance | path

Searched refs:transposed_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc142 std::vector<int64_t> transposed_shape(shape.begin(), shape.end()); in createTransposeOp() local
143 int64_t r = transposed_shape[dims - 1]; in createTransposeOp()
144 int64_t c = transposed_shape[dims - 2]; in createTransposeOp()
146 transposed_shape[dims - 1] = c; in createTransposeOp()
147 transposed_shape[dims - 2] = r; in createTransposeOp()
150 RankedTensorType::get(transposed_shape, value_type.getElementType()); in createTransposeOp()
Deinsum.cc61 SmallVector<int64_t, 4> transposed_shape(shape.begin(), shape.end()); in createTransposeOp() local
63 transposed_shape[i] = shape[permutation[i]]; in createTransposeOp()
66 RankedTensorType::get(transposed_shape, value_type.getElementType()); in createTransposeOp()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlower_general_dot.cc81 llvm::SmallVector<int64_t, 5> transposed_shape; in TransposeReshape() local
83 transposed_shape.push_back(arg_shape[val]); in TransposeReshape()
85 auto transpose_type = RankedTensorType::get(transposed_shape, element_type); in TransposeReshape()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dqr_op_impl.h178 TensorShape transposed_shape = input.shape(); in ComputeAsync() local
179 transposed_shape.set_dim(ndims - 2, input.dim_size(ndims - 1)); in ComputeAsync()
180 transposed_shape.set_dim(ndims - 1, input.dim_size(ndims - 2)); in ComputeAsync()
185 transposed_shape, &input_transposed), in ComputeAsync()
Deinsum_op_impl.h342 TensorShape transposed_shape; in TransposeOperand() local
344 transposed_shape.AddDim(input.dim_size(permutation[i])); in TransposeOperand()
349 return CopyFrom(input, transposed_shape, output); in TransposeOperand()
352 ctx->allocate_temp(DataTypeToEnum<T>::value, transposed_shape, output)); in TransposeOperand()
/external/tensorflow/tensorflow/lite/kernels/
Dbatch_matmul.cc391 RuntimeShape transposed_shape(GetTensorShape(tensor_in)); in TransposeRowsColumnsImpl() local
402 transposed_shape.SetDim(rank - 1, shape.Dims(rank - 2)); in TransposeRowsColumnsImpl()
403 transposed_shape.SetDim(rank - 2, shape.Dims(rank - 1)); in TransposeRowsColumnsImpl()
404 optimized_ops::Transpose(params, shape, input, transposed_shape, output); in TransposeRowsColumnsImpl()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc458 TensorShape transposed_shape = input.shape(); in TransposeAndConjugateTensor() local
459 transposed_shape.set_dim(input.dims() - 1, in TransposeAndConjugateTensor()
461 transposed_shape.set_dim(input.dims() - 2, in TransposeAndConjugateTensor()
464 ctx->allocate_temp(DataTypeToEnum<T>::value, transposed_shape, output)); in TransposeAndConjugateTensor()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2640 llvm::SmallVector<int64_t, 4> transposed_shape; in Verify() local
2642 transposed_shape.push_back(input_type.getDimSize(axis)); in Verify()
2645 RankedTensorType::get(transposed_shape, input_type.getElementType()); in Verify()