Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for num_elements (0.14 sec)

  1. tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc

          // zero reserved elements
          %num_elements = "tf.Const"() <{value = dense<0> : tensor<i32>}> {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : () -> tensor<i32>
    
          %list = "tf.TensorListReserve"(%elem_shape, %num_elements) : (tensor<i32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<64x1xbf16>>>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 23:59:33 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        Type shape_dtype = getElementTypeOrSelf(op.getElementShape().getType());
        Value num_elements = operands[1];
        IntegerAttr attr;
        if (matchPattern(num_elements, m_Constant(&attr))) {
          return CreateI32SplatConst(op.getLoc(), rewriter, {1}, attr.getInt());
        }
        if (auto const_op = num_elements.getDefiningOp<TF::ConstOp>()) {
          return CreateI32SplatConst(op->getLoc(), rewriter, {1},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

                                        builder->getBoolAttr(false));
    }
    
    // Creates a mhlo.rng_uniform op with `builder` to generate `num_elements`
    // 32-bit integer numbers in the range of [`lower_limit`, `upper_limit`).
    static mhlo::RngOp CreateRngUniform32(Location loc, int num_elements,
                                          int lower_limit, int upper_limit,
                                          OpBuilder *builder) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tests/ops.mlir

    }
    
    // -----
    
    func.func @testFullyConnectedWithBadInputShape(%arg0: tensor<2x2x11xf32>, %arg1: tensor<40x40xf32>, %arg2: none) -> tensor<40xf32> {
      // expected-error @+1 {{expect 'input' num_elements % 40 == 0, got input type 'tensor<2x2x11xf32>'}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 189.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

    In the current implementation, the resulting operations are statically shaped,
    which means it must be possible to infer a bound on the full shape of the
    TensorList. That is, the `element_shape` and `num_elements` arguments to a
    tensor list creation op are constant.
    
    A tensor list creation op `tf.EmptyTensorList`/`tf.TensorListReserve` will be
    turned in to a zero-initialized buffer, and the size is initialized to 0
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    input_handle: the input list
    tensor: the gathered result
    num_elements: optional. If not -1, the number of elements in the list.
      }];
    
      let arguments = (ins
        TF_VariantTensor:$input_handle,
        TF_Int32Tensor:$element_shape,
    
        DefaultValuedOptionalAttr<I64Attr, "-1">:$num_elements
      );
    
      let results = (outs
        TF_Tensor:$tensor
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  7. staging/src/k8s.io/apiserver/pkg/registry/generic/registry/store_test.go

    	if err == nil || !errors.IsAlreadyExists(err) {
    		t.Error("Expected already exists error")
    	}
    }
    
    func TestNewCreateOptionsFromUpdateOptions(t *testing.T) {
    	f := fuzz.New().NilChance(0.0).NumElements(1, 1)
    
    	// The goal here is to trigger when any changes are made to either
    	// CreateOptions or UpdateOptions types, so we can update the converter.
    	for i := 0; i < 20; i++ {
    		in := &metav1.UpdateOptions{}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Jun 12 10:12:02 UTC 2024
    - 101.8K bytes
    - Viewed (0)
Back to top