- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for RestoreCollapsedDimensions (0.22 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
// `XlaGatherOp (https://www.tensorflow.org/xla/operation_semantics#gather). The // resulting type's shape has `tensor_type.size() + collapsed_dims.size()` // dimensions. RankedTensorType RestoreCollapsedDimensions( const RankedTensorType tensor_type, const absl::flat_hash_set<int64_t>& collapsed_dims) { ArrayRef<int64_t> original_tensor_shape = tensor_type.getShape(); const int output_tensor_rank =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0)