- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for output_shapes (0.4 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
} auto output_shape = mlir::cast<RankedTensorType>(conv_op.getResult().getType()) .getShape(); SmallVector<int64_t, 4> transposed_output_shape = { output_shape[dnums.getOutputBatchDimension()], output_shape[dnums.getOutputSpatialDimensions().data()[0]], output_shape[dnums.getOutputSpatialDimensions().data()[1]],
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/transforms/optimize.cc
mlir::cast<ShapedType>(input.getType()).getShape(); ArrayRef<int64_t> output_shape = mlir::cast<ShapedType>(output.getType()).getShape(); int64_t agg_value = 1; for (size_t i = agg_start_idx; i < input_shape.size() - 1; ++i) { agg_value *= input_shape[i]; } return (agg_value == output_shape[agg_start_idx]); } // Returns whether the given type `a` is broadcast-compatible with `b`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
) ] ) ) self.assertTrue( self._contains_op( output_graphdef, 'Const', '_output_shapes', per_channel_size_attr, ) ) elif target_opset == quant_opts_pb2.UNIFORM_QUANTIZED: self.assertTrue(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
.getScale(), /*filter_scales=*/filter_quantized_type.getScales()); const ArrayRef<int64_t> output_shape = op->getResult(0).getType().cast<TensorType>().getShape(); const SmallVector<int64_t, 1> bias_shape = { output_shape[output_shape.size() - 1]}; // `tfl.fully_connected`'s `GetChannelDimIndex` is 0. const auto bias_quantized_type =
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/ir/tfl_ops.td
let summary = "Transpose convolution operator"; let description = [{ Performs transpose convolution operation on input. }]; let arguments = (ins TFL_I32Tensor:$output_shape, TFL_TensorOf<[F32, QI8, QUI8, QI16]>:$weights, TFL_TensorOf<[F32, QI8, QUI8, QI16]>:$input, TFL_TensorOfOrNone<[F32, QI32, I64]>:$bias, TFL_PaddingAttr:$padding,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
IsWithinInt32Range); if (elements_all_in_int32_range) { std::vector<int32_t> output_shape(output_ty.getRank()); std::transform(output_ty.getShape().begin(), output_ty.getShape().end(), output_shape.begin(), [](int64_t val) { return static_cast<int32_t>(val); }); output_int_type = tensorflow::GetTypeFromTFTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)