Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 72 for casts (0.06 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. pilot/pkg/networking/core/cluster_test.go

    func TestBuildGatewayClustersWithRingHashLb(t *testing.T) {
    	cases := []struct {
    		name             string
    		ringSize         int
    		expectedRingSize int
    	}{
    		{
    			"default",
    			0,
    			1024,
    		},
    		{
    			"ring size",
    			2,
    			2,
    		},
    		{
    			"use ring hash",
    			2,
    			2,
    		},
    	}
    	for _, tt := range cases {
    		t.Run(tt.name, func(t *testing.T) {
    			g := NewWithT(t)
    
    Registered: Fri Jun 14 15:00:06 UTC 2024
    - Last Modified: Thu Jun 13 01:56:28 UTC 2024
    - 108.8K bytes
    - Viewed (0)
  9. 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)
  10. pilot/pkg/networking/core/cluster_builder_test.go

    			},
    		},
    	}
    
    	cases := []struct {
    		name         string
    		expectedName string
    		expectedPath string
    	}{
    		{
    			name:         "uds",
    			expectedName: security.SDSExternalClusterName,
    			expectedPath: security.CredentialNameSocketPath,
    		},
    	}
    	for _, tt := range cases {
    		t.Run(tt.name, func(t *testing.T) {
    Registered: Fri Jun 14 15:00:06 UTC 2024
    - Last Modified: Thu Jun 13 01:56:28 UTC 2024
    - 110.4K bytes
    - Viewed (0)
Back to top