Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetTensorSpatialDimIndex (0.6 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

                                                     dim_value);
        };
    
        for (auto i : llvm::seq<int>(0, num_spatial_dims)) {
          const int64_t dim = GetTensorSpatialDimIndex(num_dims, data_format, i);
          spatial_dim_indices.push_back(dim);
    
          const int64_t dilation = get_int(dilations[dim]);
          rhs_dilations.push_back(dilation);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      if (input_ty.hasRank() && filter_ty.hasRank()) {
        // Checks the size of each of the output spatial dimensions.
        for (auto i : llvm::seq<int>(0, num_spatial_dims)) {
          const int64_t dim = GetTensorSpatialDimIndex(num_dims, format, i);
          int64_t stride = get_int(strides[dim]);
          int64_t expected_output_size;
          int64_t pad_low;
          int64_t pad_high;
          // Retrieve padding, if defined explicitly.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
Back to top