- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 232 for stride_w (0.18 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
return failure(); // We also need constant begin/end indices and strides to perform padding // calculations. // Bounded shape after performing strided slice SmallVector<int64_t, 4> shape; // Bounded begin, end, and strides for strided slice SmallVector<int64_t, 4> begin_indices, end_indices, strides; if (!op.GetSlicedShapeAndBoundRanges(&shape, &begin_indices, &end_indices,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (1) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
TFLOp op) { TfLitePoolParams pool_params; pool_params.stride_height = op.stride_h().getSExtValue(); pool_params.stride_width = op.stride_w().getSExtValue(); pool_params.filter_height = op.filter_h().getSExtValue(); pool_params.filter_width = op.filter_w().getSExtValue(); const auto padding = GetTflitePadding(inst, op.padding());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (2) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir
// CHECK-NEXT: %3 = mhlo.reshape %2 : (tensor<1xi32>) -> tensor<i32> // CHECK-NEXT: %4 = mhlo.convert %3 : tensor<i32> // CHECK-NEXT: %5 = "mhlo.slice"(%1) <{limit_indices = dense<2> : tensor<1xi64>, start_indices = dense<1> : tensor<1xi64>, strides = dense<1> : tensor<1xi64>}> : (tensor<4xi32>) -> tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// Dynamic stride: when `begin` and `end` inputs are unknown at compile time, // `strides` must be known. // CHECK: tf.StridedSlice
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
return failure(); } // Constructs strides array. // For example, [2, 3] -> [1, 2, 3, 1]. SmallVector<int64_t, num_spatial_dims + 2> strides({1}); for (const auto v : conv_op.getWindowStrides().value().getValues<int64_t>()) { strides.emplace_back(v); } strides.emplace_back(1); // Constructs dilation array.
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/lift_quantizable_spots_as_functions.td
def LiftConv : Pat< (TF_Conv2DOp:$res $input, $filter, $strides, $use_cudnn_on_gpu, $padding, $explicit_paddings, IsDataFormatNHWC:$data_format, $dilations), (LiftAsTFPartitionedCall<"composite_conv2d_fn"> (ArgumentList $input, $filter), (ResultList $res), (NamedAttributeList (NamedAttr<"strides"> $strides), (NamedAttr<"use_cudnn_on_gpu"> $use_cudnn_on_gpu),
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/cc/gradients/nn_grad.cc
string data_format; string padding; std::vector<int32> strides; bool use_cudnn_on_gpu; auto attrs = op.output(0).node()->attrs(); TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "data_format", &data_format)); TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "padding", &padding)); TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "strides", &strides)); TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "use_cudnn_on_gpu", &use_cudnn_on_gpu));
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/cc/gradients/nn_grad_test.cc
const std::vector<int> strides{1, 1, 1, 1}; auto y = Conv2D(scope_, x, filter, strides, "SAME"); RunTest(x, shape, y, shape); } TEST_F(NNGradTest, MaxPoolGradHelper) { TensorShape x_shape({1, 2, 2, 1}); TensorShape y_shape({1, 1, 1, 1}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); // Setup window and strides so that we only do one MaxPool.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 22 20:45:22 UTC 2022 - 15K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.td
def LiftConv : Pat< (TF_Conv2DOp:$res $input, $filter, $strides, $use_cudnn_on_gpu, $padding, $explicit_paddings, IsDataFormatNHWC:$data_format, $dilations), (LiftAsTFPartitionedCall<"composite_conv2d_fn"> (ArgumentList $input, $filter), (ResultList $res), (NamedAttributeList (NamedAttr<"strides"> $strides), (NamedAttr<"use_cudnn_on_gpu"> $use_cudnn_on_gpu),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/fused_kernel_matcher.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.2K bytes - Viewed (0)