Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for set_type (0.16 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

    // -----
    
    // CHECK-LABEL: func @fft_1D
    func.func @fft_1D(%arg0: tensor<8xcomplex<f32>>) -> tensor<8xcomplex<f32>> {
      // CHECK: "mhlo.fft"(%arg0) <{fft_length = dense<8> : tensor<1xi64>, fft_type = #mhlo<fft_type FFT>}> : (tensor<8xcomplex<f32>>
      %0 = "tf.FFT"(%arg0) : (tensor<8xcomplex<f32>>) -> tensor<8xcomplex<f32>>
      func.return %0 : tensor<8xcomplex<f32>>
    }
    
    // -----
    
    // CHECK-LABEL: func @ifft_1D
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        RankedTensorType act_type = mlir::dyn_cast<RankedTensorType>(act.getType());
        if (!act_type) return failure();
        Type act_ele_type = act_type.getElementType();
        // To support mixed precision, the statistics type, which maybe more
        // precise than the input types, are used for this op.
        Type kernel_type = mlir::cast<TensorType>(scale.getType()).getElementType();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  3. src/cmd/compile/internal/ssagen/ssa.go

    		base.Fatalf("deferStructFnField is %q, not fn", name)
    	}
    
    	n := ir.NewDeclNameAt(src.NoXPos, ir.OTYPE, ir.Pkgs.Runtime.Lookup("_defer"))
    	typ := types.NewNamed(n)
    	n.SetType(typ)
    	n.SetTypecheck(1)
    
    	// build struct holding the above fields
    	typ.SetUnderlying(types.NewStruct(fields))
    	types.CalcStructSize(typ)
    
    	deferType = typ
    	return typ
    }
    
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon Jun 10 19:44:43 UTC 2024
    - 284.9K bytes
    - Viewed (0)
Back to top