- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 108 for casts (0.36 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
auto input_args_have_static_shape = [&]() -> bool { return mlir::cast<TensorType>(input_tensor.getType()).hasStaticShape() && mlir::cast<TensorType>(kernel_tensor.getType()).hasStaticShape() && mlir::cast<TensorType>(window_strides.getType()).hasStaticShape() && mlir::cast<TensorType>(padding.getType()).hasStaticShape() && mlir::cast<TensorType>(lhs_dilation.getType()).hasStaticShape() &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
} // Get the tensor types. const auto input_type = input_tensor.getType().cast<ShapedType>(); const auto weights_type = weights_tensor.getType().cast<ShapedType>(); const auto bias_type = has_bias ? bias_tensor.getType().cast<ShapedType>() : ShapedType{}; const auto output_type = getType(0).cast<ShapedType>(); // Folding only implemented for float tensors.
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/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
Operation* filter_op = filter_value.getDefiningOp(); auto filter_uniform_quantized_type = GetElementType(filter_value).cast<UniformQuantizedPerAxisType>(); auto filter_constant_value_attr = cast<DenseIntElementsAttr>( cast<stablehlo::ConstantOp>(filter_value.getDefiningOp()).getValue()); const DenseIntElementsAttr new_filter_value_attr =
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/tensorflow/ir/tf_ops_n_z.cc
} } LogicalResult SpaceToBatchNDOp::verify() { SpaceToBatchNDOp op = *this; const auto input_type = op.getInput().getType().cast<TensorType>(); const auto block_shape_type = op.getBlockShape().getType().cast<TensorType>(); const auto paddings_type = op.getPaddings().getType().cast<TensorType>(); // Check that block_shape has rank 1. if (!IsOfRankOrUnranked(op.getBlockShape(), 1)) {
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/mlir/tensorflow/ir/tf_ops_a_m.cc
// types match the result types. auto yield = cast<YieldOp>(region.front().getTerminator()); auto updated_results = llvm::to_vector<4>(yield.getOperands()); // If the yield types do not match the IfRegion result types, add appropriate // casts. rewriter.setInsertionPoint(yield); for (auto it : llvm::zip(op.getResultTypes(), updated_results)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
// If there is a change in supporting more types in the TFLite cast op kernel, // the While loop outline pass should be updated since it inserts cast op(s) // after the TF -> TFL legalization pass is done. // LINT.IfChange def TFL_CastOp : TFL_Op<"cast", [ Pure, SameOperandsAndResultShape]> { let summary = "Cast operator"; let description = [{ Casts input from input type to output type. }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
maven-core/src/test/resources/apiv4-repo/org/hamcrest/hamcrest-core/1.3/hamcrest-core-1.3-javadoc.jar
TypeSafeDiagnosingMa<T> extends BaseMatcher<T> Convenient base class for Matchers that require a non-null value of a specific type and that will report why the received value has been rejected. This implements the null check, checks the type and then casts. To use, implement matchesSafely() . Author: Neil Dunn, Nat Pryce, Steve Freeman Constructor Summary protected TypeSafeDiagnosingMa() The default constructor for simple sub types protected TypeSafeDiagnosingMa(java.lang.Class<?> expectedType) Use this...
Registered: Wed Jun 12 09:55:16 UTC 2024 - Last Modified: Sun Oct 02 08:41:25 UTC 2022 - 236.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
bool IsLastDimEqualToNumElements(Type type1, Type type2) { return (mlir::cast<ShapedType>(type1).getRank() >= 1 && mlir::cast<ShapedType>(type1).getDimSize( mlir::cast<ShapedType>(type1).getRank() - 1) == mlir::cast<ShapedType>(type2).getNumElements()); } bool CanFuseConvOrDepthwiseConvShapes(const ArrayRef<int64_t> filter_shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
pilot/pkg/networking/core/cluster_test.go
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Thu Jun 13 01:56:28 UTC 2024 - 108.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK-SAME: (tensor<*xf32>) -> tensor<*xi8> // CHECK: %[[TENSOR_CAST:.*]] = tensor.cast %[[CAST_RESULT_2]] : tensor<*xi8> to tensor<?x?x?x?xi8> %2 = tensor.cast %28 : tensor<*xi8> to tensor<?x?x?x?xi8> // CHECK: %[[CAST_RESULT_0:.*]] = "tf.Cast"(%arg0) // CHECK-SAME: (tensor<16x194x199x4xf32>) -> tensor<16x194x199x4xui8> // CHECK: %[[CAST_RESULT_1:.*]] = "tf.Cast"(%arg0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0)