- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetPermutationFromTrivialReshape (0.25 sec)
-
tensorflow/compiler/mlir/lite/utils/utils.td
"$1.getType().cast<ShapedType>())">>; // Returns the permutation of the trivial reshape op, this will be used to // construct the transpose op. def GetPermutationFromTrivialReshape : NativeCodeCall< "TFL::GetPermutationFromTrivialReshape(" "$0.getType().cast<ShapedType>()," "$1.getType().cast<ShapedType>())">; // Constraint that checks if all values in offset between two // attributes are non-negative.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_patterns.td
(TF_TransposeOp:$transpose_out1 $input, (Arith_ConstantOp:$permutation1 $p1)), $_), (TF_TransposeOp:$transpose_op $input, (Arith_ConstantOp (RemapPermutation $permutation1, (GetPermutationFromTrivialReshape $transpose_out1, $output)))), [(IsReshapeEquivalentToTranspose $transpose_out1, $output), (AnyStaticShapeTensor $input), (AnyStaticShapeTensor $output), (HasOneUse $transpose_out1)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
} // Returns the permutation that maps the input shape to the output shape. // This is only valid for trivial reshape ops. inline DenseElementsAttr GetPermutationFromTrivialReshape( ShapedType input_type, ShapedType output_type) { ArrayRef<int64_t> in_shape = input_type.getShape(); ArrayRef<int64_t> out_shape = output_type.getShape();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0)