- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 117 for set_shape (0.13 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
auto new_shape_const_attr = DenseElementsAttr::get(shape_spec_type, new_shape.getShape()); rewriter.setInsertionPointAfter(weight_op); auto new_shape_const = rewriter.create<TF::ConstOp>( weight_op->getLoc(), shape_spec_type, new_shape_const_attr); auto reshape_op = rewriter.create<TF::ReshapeOp>( weight_op->getLoc(), new_shape, weight_op->getResult(0), new_shape_const);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
// Thus, we fail to match if the consuming reshape rank is larger. ArrayRef<int64_t> input_shape = input_type.getShape(); if (reshape_shape.size() > input_shape.size()) return failure(); // Extend the input shape with leading 1s to match the broadcast shape. ArrayRef<int64_t> broadcast_shape = output_type.getShape(); SmallVector<int64_t, 4> input_shape_extended;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
auto output_types = fused_lstm_func_.getFunctionType().getResults(); SmallVector<int64_t, 2> output_shape{1, mlir::ShapedType::kDynamic}; EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getShape().size(), output_shape.size()); for (int i = 0; i < output_shape.size(); i++) { EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getDimSize(i), output_shape[i]); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
Value temp_padding = CreateConstValue<int32_t>(builder, loc, {num_dims, 2}, padding_values); SmallVector<int64_t> output_shape(input_shape.getShape().begin(), input_shape.getShape().end()); for (int i : spatial_dims) { output_shape[i] += padding_values[2 * i] + padding_values[2 * i + 1]; } return builder.create<TF::PadV2Op>(
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/optimize_batch_matmul.cc
bmm_op->getLoc(), permuation_tensor_type, DenseElementsAttr::get(permuation_tensor_type, permute)); auto input_shape = input_type.getShape(); llvm::SmallVector<int64_t, 4> permuted_shape(input_shape.begin(), input_shape.end()); // Swaps z dimension and x dimension to get permuted shape.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.h
// ShapeContainerT is any type with the following methods: // bool hasRank() // ArrayRef<int64_t> getShape() // This includes mlir::TF::ShapeAttr and mlir::ShapedType. template <typename ShapeContainerT> void SetTensorShapeProto(ShapeContainerT shape, TensorShapeProto* proto) { if (shape.hasRank()) { for (int64_t dim : shape.getShape()) { proto->add_dim()->set_size(mlir::ShapedType::isDynamic(dim) ? -1 : dim); } } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc
} ApiConverter::ToC(old_shape, &old_shape_c); executor->TpuTransferManager_GetInfeedLayoutFn(&old_shape_c, &new_shape_c); xla::Shape new_shape = ApiConverter::FromC(&new_shape_c); ApiConverter::Destroy(&old_shape_c); ApiConverter::Destroy(&new_shape_c); auto minor_to_major = new_shape.layout().minor_to_major(); return std::vector<int64_t>(minor_to_major.begin(), minor_to_major.end()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/c/ops_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status)); TF_ShapeHandle* new_shape = TF_ShapeInferenceContextVectorFromSize( ctx, TF_ShapeInferenceContextRank(ctx, handle)); TF_ShapeInferenceContextSetOutput(ctx, 0, new_shape, status); TF_DeleteShapeHandle(handle); TF_DeleteShapeHandle(new_shape); } TEST(OpsTest, TestShapeInference_VectorizeFunction) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.h
return operand_ty == result_ty && identity_ty.hasStaticShape() && result_ty.hasStaticShape() && OpTrait::util::staticallyKnownBroadcastable(operand_ty.getShape(), identity_ty.getShape()); }; // Check that we have a constant operand on one side (candidate for identity). const bool is_commutative =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// Reshape input image to add a new spatial dimension. auto image_type = mlir::cast<ShapedType>(conv_op.getLhs().getType()); SmallVector<int64_t, 4> image_2d_shape(image_type.getShape().begin(), image_type.getShape().end()); image_2d_shape.push_back(1); auto image_2d_type = RankedTensorType::get(image_2d_shape, image_type.getElementType()); auto loc = conv_op.getLoc();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)