- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 88 for input_shapes_ (0.49 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/decompose_optionals.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize.mlir
func.func private @conv(%input: tensor<1x3x4x3xf32> {tf._user_specified_name = "input_tensor"}) -> tensor<*xf32> attributes {tf._construction_context = "kEagerRuntime", tf._input_shapes = [#tf_type.shape<1x3x4x3>]} { %weight = arith.constant dense_resource<__elided__> : tensor<2x3x3x2xf32> %bias = arith.constant dense<[7.11401462, 7.05456924]> : tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
Builder* builder) { const auto& input_dtype = input_tensor.dtype(); const auto& input_shape = input_tensor.shape(); Type elt_type; TF_RETURN_IF_ERROR(ConvertDataType(input_dtype, *builder, &elt_type)); SmallVector<int64_t, 4> shape; ConvertToMlirShape(input_shape, &shape); auto type = RankedTensorType::get(shape, elt_type); #define CONVERT_FLAT(DTYPE, CTYPE) \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
ArrayRef<int64_t> input_shape = input_tensor.getShapedType().getShape(); auto output_type = mlir::cast<ShapedType>(op.getOutput().getType()); SmallVector<int32_t, 4> perm; SmallVector<int64_t, 4> output_shape; for (int i = 0; i < num_dimensions; ++i) { perm.push_back(perm_tensor.getValues<IntegerAttr>()[i].getInt()); output_shape.push_back(input_shape[perm[i]]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.cc
return errors::InvalidArgument(absl::StrCat( "Length of input node array and data shape doesn't match (#arrays ", node_names.size(), ", #input_shapes ", node_shapes.size(), ")")); } // StringMap doesn't support reserve else reserve input map size here. for (int i = 0, end = node_names.size(); i < end; i++) { auto& name = node_names[i];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
auto dim_attr = DenseIntElementsAttr::get(dim_type, dims); auto dim = rewriter.create<arith::ConstantOp>(op.getLoc(), dim_attr); input_shape.insert(input_shape.begin() + dim_to_expand, 1); TensorType expanded_type = input_type.clone(input_shape); input = rewriter.create<TFL::ExpandDimsOp>(op.getLoc(), expanded_type, input, dim);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
// the input tensor's dimensions, return 0-valued tensor of the requested // shape. ArrayRef<int64_t> input_shape = GetRankedTensorShape(input); for (int i = 0, end = input_shape.size(); i < end; i++) { if (begin_values[i] < 0 || (begin_values[i] + size_values[i] > input_shape[i])) { return CreateF32SplatConst(builder, size_shape, 0, location); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
int num_input_dims = ranked_input_type.getRank(); SmallVector<int32_t, 4> padding_begin(num_input_dims, 0); auto input_shape = ranked_input_type.getShape(); SmallVector<int32_t, 4> padding_end(input_shape.begin(), input_shape.end()); SmallVector<int32_t, 4> padding_strides(num_input_dims, 1); int begin_mask = strided_slice_op.getBeginMask();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops.mlir
// Test hoist var handle op and read variable op in the batch function. // CHECK-LABEL: func private @batched_function func.func private @batched_function(%arg0: tensor<1x3xf32>) -> tensor<1x3xf32> attributes {tf._input_shapes = [#tf_type.shape<1x3>, #tf_type.shape<*>], tf.signature.is_stateful} { // CHECK-NOT: tf.VarHandleOp // CHECK-NOT: tf.ReadVariableOp // CHECK: "tf._TfrtGetResource"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 01 23:54:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
// Graphdef import path. module = tensorflow::LoadFromGraphdefOrMlirSource( input_file_name, input_mlir, use_splatted_constant, custom_opdefs, specs, debug_info_file, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, &source_mgr, &context); } // If errors occur, the library call in the above already logged the error // message. So we can just return here.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 14K bytes - Viewed (0)