Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 36 for input_size (0.15 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc

          scalar_mul(dilation_rate_value, scalar_sub(filter_size, one)), one);
    
      // output_size = (input_size + stride - 1) / stride
      Value output_size = scalar_div(
          scalar_add(input_size, scalar_sub(stride_value, one)), stride_value);
      // padding_needed = std::max(
      //     0,
      //     (output_size - 1) * stride + effective_filter_size - input_size)
      Value padding_needed = scalar_sub(
          scalar_add(effective_filter_size_op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/layout_optimization_layout_assignment_gpu_cc_70.mlir

    func.func @transposeConv2DBackpropInput_f32(
      %input_size:   tensor<4xi32>,
      %filter:       tensor<1x28x28x64xf32>,
      %out_backprop: tensor<1x28x28x64xf32>
    ) -> tensor<1x28x28x64xf32> {
    
      // CHECK: "tf.Conv2DBackpropInput"
      // CHECK-SAME: data_format = "NCHW"
      %0 = "tf.Conv2DBackpropInput"(%input_size, %filter, %out_backprop)
           {
             data_format = "NHWC",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 21 08:41:18 UTC 2022
    - 8.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/layout_optimization_layout_assignment_gpu_cc_60.mlir

    func.func @transposeConv2DBackpropInput_f16(
      %input_size:   tensor<4xi32>,
      %filter:       tensor<1x28x28x64xf16>,
      %out_backprop: tensor<1x28x28x64xf16>
    ) -> tensor<1x28x28x64xf16> {
    
      // CHECK: "tf.Conv2DBackpropInput"
      // CHECK-SAME: data_format = "NCHW"
      %0 = "tf.Conv2DBackpropInput"(%input_size, %filter, %out_backprop)
           {
             data_format = "NHWC",
             padding = "VALID",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 21 08:41:18 UTC 2022
    - 5.8K bytes
    - Viewed (0)
  4. tensorflow/c/c_test_util.cc

          }
        }
      }
      return found_dtype && found_value;
    }
    
    bool IsAddN(const tensorflow::NodeDef& node_def, int n) {
      if (node_def.op() != "AddN" || node_def.name() != "add" ||
          node_def.input_size() != n) {
        return false;
      }
      bool found_t = false;
      bool found_n = false;
      for (const auto& attr : node_def.attr()) {
        if (attr.first == "T") {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

            int64_t output_size;
            int64_t pad_low_int64;
            int64_t pad_high_int64;
            int64_t input_size = input_ty.getDimSize(dim);
            if (input_size == ShapedType::kDynamic) return failure();
            absl::Status status = tensorflow::GetWindowedOutputSizeVerbose(
                input_size, filter_ty.getDimSize(i), dilation, stride, padding,
                &output_size, &pad_low_int64, &pad_high_int64);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        Location loc = op.getLoc();
        Value scalar_zero = CreateI32SplatConst(loc, &rewriter, {}, 0);
    
        // Compute the input tensorlist's length and store it in `input_size`.
        IntegerType shape_dtype = rewriter.getIntegerType(32);
        auto input_size = rewriter.create<TF::TensorListLengthOp>(
            loc, tensorflow::GetTypeFromTFTensorShape({}, shape_dtype),
            op.getOperand(0));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  7. tensorflow/cc/tools/freeze_saved_model.cc

    StatusOr<string> GetVarHandleName(
        const std::unordered_map<string, NodeDef*>& name_to_node_map,
        string node_name) {
      const NodeDef* node = name_to_node_map.at(node_name);
      while (node->input_size() > 0) {
        auto parent = name_to_node_map.find(node->input(0));
        if (parent == name_to_node_map.end()) break;
        node = parent->second;
        if (node->op() != "Identity") {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 11 08:05:36 UTC 2023
    - 11.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

          for (int64_t i = 0; i < input_ty.getRank(); ++i) {
            int64_t slice_size = slice_sizes.getValues<APInt>()[i].getSExtValue();
            int64_t input_size = input_shape[i];
            if (slice_size != -1 && input_size != ShapedType::kDynamic &&
                slice_size > input_size) {
              return op.emitOpError() << "requires size[i] <= Di, even if begin[i] "
                                         "is unknown at compile time";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

                            std::vector<const Edge*> control_deps, Output* size) {
      // If slice_size[i] >= 0 then slice_size[i] = slice_size[i].
      //
      // If slice_size[i] == -1 then slice_size[i] = input_size[i] -
      // begin[i].
      //
      // If slice_size[i] < -1 then executing the slice will throw an error, and we
      // don't do anything here.  We've already filtered these cases out in
      // IsRewritableSlice.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

                               b.device());
        }
        return false;
      }
      if (a.input_size() != b.input_size()) {
        if (diff) {
          *diff = absl::StrCat(diff_preamble, " mismatch for node ", a.name(),
                               ", expected ", a.input_size(), " inputs got ",
                               b.input_size(), " expected:\n", a.DebugString(),
                               "\ngot:\n", b.DebugString());
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
Back to top