- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for spatial_dims (0.18 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
Value &padding, int num_dims) { ShapedType input_shape = mlir::cast<ShapedType>(input.getType()); SmallVector<int64_t> spatial_dims(num_dims - 2); absl::c_iota(spatial_dims, 1); bool has_dynamic_spatial_dim = absl::c_any_of( spatial_dims, [&input_shape](int64_t dim) { return input_shape.isDynamicDim(dim); }); if (conv_padding.strref() == "SAME" && has_dynamic_spatial_dim) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// Prepare metadata indexed by spatial_dim for computing pad_before // and pad_after. int64_t input_size = input_shape[spatial_dim]; if (input_size == ShapedType::kDynamic) return failure(); int64_t output_size = out_backprop_ty.getDimSize(spatial_dim); if (output_size == ShapedType::kDynamic) return failure(); int64_t filter_size = filter_ty.getDimSize(i);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)