Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getFilterSizes (0.2 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      // Permute filter sizes operand.
      OpBuilder builder(getOperation());
      auto filter_sizes_permuted = builder.create<TF::DataFormatVecPermuteOp>(
          getLoc(), getFilterSizes(),
          StringAttr::get(getContext(), src_data_format),
          StringAttr::get(getContext(), data_format));
      setOperand(1, filter_sizes_permuted);
    
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        ArrayRef<int64_t> out_backprop_shape = out_backprop_ty.getShape();
        ArrayRef<int64_t> input_shape = input_ty.getShape();
    
        DenseIntElementsAttr filter_shape_attr;
        if (!matchPattern(op.getFilterSizes(), m_Constant(&filter_shape_attr)) ||
            filter_shape_attr.getType().getRank() != 1)
          return failure();
    
        auto dilations_attr = GetI64ElementsAttr(op.getDilations());
        std::vector<int> dilations{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top