Searched defs:permuted_shape (Results 1 – 7 of 7) sorted by relevance
50 arm_compute::TensorShape permuted_shape = shape; in compute_permutation_parameters() local274 TensorShape permuted_shape = tensor.info()->tensor_shape(); in access_tensor() local351 TensorShape permuted_shape = tensor.info()->tensor_shape(); in access_tensor() local
405 TensorShape permuted_shape = tensor.info()->tensor_shape(); in fill_tensor() local
111 …const TensorShape permuted_shape = misc::shape_calculator::compute_permutation_output_sh… in validate() local
144 …const TensorShape permuted_shape = misc::shape_calculator::compute_permutation_output_sh… in validate() local
289 bool permuted_shape = false; in check_requires_permute() local
940 auto permuted_shape = Shape(scope, permuted); in ProdGrad() local
997 Shape permuted_shape = ShapeUtil::PermuteDimensions(permutation, shape()); in Transpose() local