Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for sizeType (0.49 sec)

  1. src/cmd/compile/internal/ssa/memcombine.go

    	if needSwap && (size != 1 || !root.Block.Func.Config.haveByteSwap(n)) {
    		return false
    	}
    
    	// This is the commit point.
    
    	// First, issue load at lowest address.
    	v = loadBlock.NewValue2(pos, OpLoad, sizeType(n*size), r[0].load.Args[0], mem)
    
    	// Byte swap if needed,
    	if needSwap {
    		v = byteSwap(loadBlock, pos, v)
    	}
    
    	// Extend if needed.
    	if n*size < root.Type.Size() {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Mar 21 19:45:41 UTC 2024
    - 18.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc

    }
    
    Value ReshapeScalarToSizeType(OpBuilder builder, Value scalar, Location loc) {
      auto size_type = GetSizeType(builder);
      return builder.create<TF::ReshapeOp>(
          loc, ArrayRef<Type>{size_type},
          ArrayRef<Value>{scalar, GetR1Const(size_type.getShape(), builder, loc)});
    }
    
    LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc

      if (!begin_type || !begin_type.hasStaticShape()) return failure();
    
      auto size = slice_op.getSize();
      auto size_type = mlir::dyn_cast_or_null<RankedTensorType>(size.getType());
      if (!size_type || !size_type.hasStaticShape()) return failure();
    
      auto output_type =
          mlir::dyn_cast_or_null<RankedTensorType>(slice_op.getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 25.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc

      void runOnOperation() final;
    };
    
    // Returns the type of the local variable for the stack size.
    Type GetSizeVarType(OpBuilder builder) {
      auto size_type = cutil::GetSizeType(builder);
      return RankedTensorType::get(
          {}, TF::ResourceType::get(ArrayRef<TensorType>{size_type},
                                    builder.getContext()));
    }
    
    // Returns the aliasing argument number of a fucntion return value if it simply
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

    struct SizeInfo {
      Value size;
      bool fixed;
    };
    
    // Modifies a function's signature to rewrite tensor list arguments to buffers
    // and sizes.
    void ModifyFunctionSignature(
        func::FuncOp func, Type size_type,
        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size,
        llvm::function_ref<std::optional<Type>(int64_t)> arg_to_buffer_type,
        llvm::function_ref<bool(int64_t)> arg_buffer_size_is_fixed) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
Back to top