- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 98 for dilation_h (1.1 sec)
-
tensorflow/compiler/mlir/tfr/examples/mnist/ops_defs.py
attrs=[ 'stride_w: int', 'stride_h: int', 'dilation_w: int', 'dilation_h: int', 'padding: {"SAME", "VALID"}', 'act: {"", "RELU", "RELU6", "TANH"} = ""' ], derived_attrs=['T: {float, int8}'], outputs=['o: T']) def _composite_conv_add_relu(input_, filter_, bias, stride_w, stride_h, dilation_w, dilation_h, padding, act): res = tf.raw_ops.Conv2D( input=input_,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 20:23:51 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/mnist_ops_test.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc
llvm::StringMap<Attribute>& identifier_to_attr) { ArrayAttr dilations = mlir::dyn_cast<ArrayAttr>(identifier_to_attr["dilations"]); const int dilation_h = mlir::cast<IntegerAttr>(dilations[1]).getInt(); const int dilation_w = mlir::cast<IntegerAttr>(dilations[2]).getInt(); return rewriter.getI64ArrayAttr({dilation_h, dilation_w}); } Attribute GetPaddingValue(PatternRewriter& rewriter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
inferredReturnTypes.assign({result_type}); return success(); } auto stride_h = op.getStrideHAttr().getInt(); auto stride_w = op.getStrideWAttr().getInt(); auto dilation_h = op.getDilationHFactorAttr().getInt(); auto dilation_w = op.getDilationWFactorAttr().getInt(); // We don't have EXPLICIT PADDING in TfLite. auto paddings = op.getPadding(); tensorflow::Padding padding;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
int dilation_i = mlir::cast<IntegerAttr>(dilations[i]).getInt(); int out_size = tflite::ComputeOutSize(kTfLitePaddingSame, input_size, filter_size, stride_i, dilation_i); int offset = 0; int padding_before = tflite::ComputePaddingWithOffset( stride_i, dilation_i, input_size, filter_size, out_size, &offset);
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/lite/transforms/dilated_conv.h
"Conv2D op doesn't have valid padding"); } // Make sure dilations are all ones if set. const ArrayAttr& dilations = op->template getAttrOfType<ArrayAttr>("dilations"); if (dilations && !TFIntListIsAllOnes(dilations)) { return rewriter.notifyMatchFailure(op, "dilations should be all 1"); } if (!TFL::TFTypeIsFloat32Tensor(op.getInput()) &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20K bytes - Viewed (0) -
tensorflow/cc/gradients/nn_grad.cc
return errors::InvalidArgument( "Conv2DBackpropInput grad requires 1 grad input"); } std::vector<int> dilations, strides, explicit_paddings; bool use_cudnn_on_gpu; std::string data_format, padding; TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "dilations", &dilations)); TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "strides", &strides)); TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 27 23:34:33 UTC 2022 - 24.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.td
(NamedAttr<"explicit_paddings"> $explicit_paddings), (NamedAttr<"dilations"> $dilations))), [(IsNotInLiftedFunc $res)], [], (addBenefit 1)>; def LiftDepthwiseConv : Pat< (TF_DepthwiseConv2dNativeOp:$res $input, $filter, $strides, $padding, $explicit_paddings, IsDataFormatNHWC:$data_format, $dilations), (LiftAsTFPartitionedCall<"composite_depthwise_conv2d_fn"> (ArgumentList $input, $filter),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// Constructs dilation array. SmallVector<int64_t, 4> dilation; if (auto rhs_dilation = conv_op.getRhsDilation()) { // For example, [2, 3] -> [1, 2, 3, 1]. dilation.emplace_back(1); dilation.append(rhs_dilation.value().getValues<int64_t>().begin(), rhs_dilation.value().getValues<int64_t>().end()); dilation.emplace_back(1); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.td
(TF_CastOp (TF_IdentityOp $filter), $truncate1), $strides, $use_cudnn, $padding, $explicit_padding, IsDataFormatNHWC:$data_format, $dilations), (CreateXLAConvOpFromTFConv2DOp $input, $filter, $input_zp, $conv, $strides, $dilations, $padding, $explicit_padding), [(IsInt8ElementType $input), (IsInt8ElementType $filter), (IsConstTensor $input_zp), (IsConstTensor $filter),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 21.1K bytes - Viewed (0)