Home
last modified time | relevance | path

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

/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.h223 TensorShape transposed_shape; in TransposeOperand() local
225 transposed_shape.AddDim(input.dim_size(permutation[i])); in TransposeOperand()
230 return CopyFrom(input, transposed_shape, output); in TransposeOperand()
233 ctx->allocate_temp(DataTypeToEnum<T>::value, transposed_shape, output)); in TransposeOperand()
/external/tensorflow/tensorflow/lite/kernels/
Dbatch_matmul.cc395 RuntimeShape transposed_shape(GetTensorShape(tensor_in)); in TransposeRowsColumnsImpl() local
406 transposed_shape.SetDim(rank - 1, shape.Dims(rank - 2)); in TransposeRowsColumnsImpl()
407 transposed_shape.SetDim(rank - 2, shape.Dims(rank - 1)); in TransposeRowsColumnsImpl()
408 optimized_ops::Transpose(params, shape, input, transposed_shape, output); in TransposeRowsColumnsImpl()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc467 TensorShape transposed_shape = input.shape(); in TransposeAndConjugateTensor() local
468 transposed_shape.set_dim(input.dims() - 1, in TransposeAndConjugateTensor()
470 transposed_shape.set_dim(input.dims() - 2, in TransposeAndConjugateTensor()
473 ctx->allocate_temp(DataTypeToEnum<T>::value, transposed_shape, output)); in TransposeAndConjugateTensor()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLGEMMDeconvolutionLayer.cpp163 TensorShape transposed_shape(reshaped_shape[1], reshaped_shape[0]); in validate() local
164 …reshaped_t_info = reshaped_info.clone()->set_is_resizable(true).set_tensor_shape(transposed_shape); in validate()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Deinsum.cc90 SmallVector<int64_t, 4> transposed_shape(shape.begin(), shape.end()); in createTransposeOp() local
92 transposed_shape[i] = shape[permutation[i]]; in createTransposeOp()
95 RankedTensorType::get(transposed_shape, value_type.getElementType()); in createTransposeOp()
Dlegalize_hlo.cc112 llvm::SmallVector<int64_t> transposed_shape(permutation_array.size()); in GetPermutationAndTransposedShape() local
114 transposed_shape[i] = input_type.getDimSize(permutation_array[i]); in GetPermutationAndTransposedShape()
117 RankedTensorType::get(transposed_shape, input_type.getElementType()); in GetPermutationAndTransposedShape()
523 llvm::SmallVector<int64_t, 4> transposed_shape(num_spatial_dims + 2); in GetReformatTypeAndPermutation() local
524 transposed_shape[default_batch_dim] = shape[batch_dim]; in GetReformatTypeAndPermutation()
525 transposed_shape[default_feature_dim] = shape[feature_dim]; in GetReformatTypeAndPermutation()
528 transposed_shape[default_spatial_dim_start + i] = in GetReformatTypeAndPermutation()
532 RankedTensorType::get(transposed_shape, type.getElementType()); in GetReformatTypeAndPermutation()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc214 std::vector<int64_t> transposed_shape = merged_shape; in Subgroup() local
221 transposed_shape.push_back(tile_assignment.dim(dim)); in Subgroup()
226 Array<int64_t> new_tiles(transposed_shape); in Subgroup()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc3370 llvm::SmallVector<int64_t, 4> transposed_shape; in verify() local
3372 transposed_shape.push_back(input_type.getDimSize(axis)); in verify()
3375 RankedTensorType::get(transposed_shape, input_type.getElementType()); in verify()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1447 llvm::SmallVector<int64_t, 2> transposed_shape = {2, input_rank}; in matchAndRewrite() local