Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for getOutBackprop (0.31 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc

      // Build new BackPropFilterOp.
      auto loc = backprop.getLoc();
      auto new_backprop = builder.create<TF::Conv2DBackpropFilterOp>(
          loc, new_result_type, input, new_filter_sizes, backprop.getOutBackprop(),
          strides, backprop.getUseCudnnOnGpu(), backprop.getPadding(),
          backprop.getExplicitPaddings(), backprop.getDataFormat(),
          backprop.getDilations());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 29.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc

          CreateCastToInt32(tf_op.getInputSizes(), op->getLoc(), rewriter);
    
      rewriter.replaceOpWithNewOp<TFL::Conv3DTransposeOp>(
          op, tf_op.getType(), output_shape, tf_op.getFilter(),
          tf_op.getOutBackprop(),
          /*bias=*/none, dilation_depth_factor, dilation_height_factor,
          dilation_width_factor,
          /*fused_activation_function=*/rewriter.getStringAttr("NONE"), padding,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

        if (!HasRankAtLeast(op.getOutBackprop(), 2))
          return op.emitOpError(
              "requires out_backprop operand to have rank at least two with `NHWC` "
              "data format");
      } else {
        // Op definition requires data_format to be either NHWC or NCHW.
        DCHECK_EQ(format, tensorflow::TensorFormat::FORMAT_NCHW);
        if (!HasRankAtLeast(op.getOutBackprop(), 3))
          return op.emitOpError(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        tensorflow::Padding padding;
        if (!GetPaddingFromString(op.getPadding().str(), &padding).ok())
          return failure();
    
        auto out_backprop_ty =
            mlir::dyn_cast<RankedTensorType>(op.getOutBackprop().getType());
        auto filter_ty = mlir::dyn_cast<RankedTensorType>(op.getFilter().getType());
    
        // With the exception of out_backprop's batch dimension, out_backprop and
    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