- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 31 for input_size (0.16 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
window_strides_values[0], input_sizes[0], output_sizes[0])) { align_corners = true; return success(); } if (can_convert_to_bilinear( /*align_corners=*/false, lhs_dilation_values[0], padding_values[0], window_strides_values[0], input_sizes[0], output_sizes[0])) { align_corners = false;
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/lite/tests/lower-static-tensor-list.mlir
// CHECK: [[ZERO_1:%.*]] = arith.constant dense<0> : tensor<i32> // CHECK: [[INPUT_SIZE:%.*]] = "tf.Gather"([[SHAPE]], [[ZERO_1]]) <{validate_indices = true}> : (tensor<2xi32>, tensor<i32>) -> tensor<i32> // CHECK: [[SIZE_DIFF:%.*]] = "tf.Sub"([[SIZE]], [[INPUT_SIZE]]) : (tensor<i32>, tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 39.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
} static LogicalResult ComputeConvWindowedOutputSize( int64_t input_size, int64_t filter_size, int64_t dilation_rate, int64_t stride, tensorflow::Padding padding, int64_t* output_size) { int64_t pad_low; int64_t pad_high; tensorflow::Status status = tensorflow::GetWindowedOutputSizeVerbose( input_size, filter_size, dilation_rate, stride, padding, output_size, &pad_low, &pad_high);
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/lite/transforms/prepare_tf.cc
mlir::dyn_cast<RankedTensorType>(input.getType()); if (!input_type) { return failure(); } const ArrayRef<int64_t> input_shape = input_type.getShape(); const int input_size = input_shape.size(); RankedTensorType begin_type = mlir::cast<RankedTensorType>(begin.getType()); const ArrayRef<int64_t> begin_shape = begin_type.getShape(); const int begin_dim = begin_shape.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0)