- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 72 for casts (0.06 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) -
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) -
pkg/config/validation/validation_test.go
}, }, valid: false, }, } for _, c := range cases { if got := validateTrafficPolicy(c.in).Err; (got == nil) != c.valid { t.Errorf("ValidateTrafficPolicy failed on %v: got valid=%v but wanted valid=%v: %v", c.name, got == nil, c.valid, got) } } } func TestValidateConnectionPool(t *testing.T) { cases := []struct { name string in *networking.ConnectionPoolSettings
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Tue Apr 30 03:11:45 UTC 2024 - 196K bytes - Viewed (0) -
pilot/pkg/networking/core/cluster_builder_test.go
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Thu Jun 13 01:56:28 UTC 2024 - 110.4K bytes - Viewed (0)