Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 10 for validate_indices (0.2 sec)

  1. tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/decompose_resource_op.mlir

    func.func @gather(%indices: tensor<?xi32>,
                 %resource: tensor<*x!tf_type.resource>) -> tensor<*xi32> {
      %0 = "tf.ResourceGather"(%resource, %indices) {batch_dims = 0 : i64, device = "/device:CPU:0", validate_indices = true}: (tensor<*x!tf_type.resource>, tensor<?xi32>) -> (tensor<*xi32>)
      func.return %0 : tensor<*xi32>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 00:40:32 UTC 2024
    - 1.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir

        %0 = "tf.VarHandleOp"() <{container = "", shared_name = "Variable"}> : () -> tensor<!tf_type.resource<tensor<2x3xbf16>>>
        %1 = "tf.ResourceGather"(%0, %cst) <{batch_dims = 0 : i64, validate_indices = true}> : (tensor<!tf_type.resource<tensor<2x3xbf16>>>, tensor<1xi32>) -> tensor<1x3xbf16>
        return %1: tensor<1x3xbf16>
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 15:33:17 UTC 2024
    - 5.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

    def LegalizeGreaterEqual : Pat<(TF_GreaterEqualOp $l, $r),
                                   (TFL_GreaterEqualOp $l, $r)>;
    
    // Gather in TF -> Gather in TFL with axis=0
    // The 'validate_indices' attribute is deprecated.
    def LegalizeGather: Pat<
      (TF_GatherOp $params, $indices, $ignored_validate_indices),
      (TFL_GatherOp $params, $indices, ConstantAttr<I32Attr, "0">,
         ConstantAttr<I32Attr, "0">)>;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td

        ]
      >;
    
    // Pattern to decompose tf.ResourceGather into tf.ReadVariable and tf.GatherV2.
    def DecomposeResourceGather : Pat<
      (TF_ResourceGatherOp:$old_result
        $resource, $indices, $batch_dims, $validate_indices),
      (TF_GatherV2Op:$dest
        (CreateTFReadVariableOp $old_result, $old_result, $resource),
        $indices,
        (TF_ConstOp $batch_dims), // axis
        $batch_dims
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 20.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir

      %0 = "tf.SparseToDense"(%arg0, %arg1, %arg2, %arg3) {validate_indices = true}: (tensor<3x5xi32>, tensor<3xi32>, tensor<2xf32>, tensor<f32>) -> tensor<?x?x?xf32>
      func.return %0 : tensor<?x?x?xf32>
      // CHECK-LABEL: sparse_to_dense_with_2d_sparse_indices_and_second_dim_greater_than_4
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 05 01:54:33 UTC 2024
    - 153.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

    // CHECK:           %9 = "tf.Gather"(%7, %cst_5) <{validate_indices = true}> : (tensor<3xi32>, tensor<2xi64>) -> tensor<2xi32>
    // CHECK-DAG:       %cst_6 = "tf.Const"() <{value = dense<1> : tensor<1xi64>}> : () -> tensor<1xi64>
    // CHECK:           %10 = "tf.Gather"(%8, %cst_6) <{validate_indices = true}> : (tensor<3xi32>, tensor<1xi64>) -> tensor<1xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

                                                  /*use_32bit=*/true_attr);
        rewriter.replaceOpWithNewOp<TF::GatherOp>(
            op, op.getType(), shape, CreateI32SplatConst(loc, &rewriter, {}, 0),
            /*validate_indices=*/true_attr);
        return success();
      }
    };
    
    struct ConvertTensorListStack
        : public OpConversionPattern<TF::TensorListStackOp> {
      using OpConversionPattern::OpConversionPattern;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/schema/schema_generated.h

      auto _validate_indices = _o->validate_indices;
      return tflite::CreateSparseToDenseOptions(
          _fbb,
          _validate_indices);
    }
    
    inline EqualOptionsT *EqualOptions::UnPack(const ::flatbuffers::resolver_function_t *_resolver) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 1M bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

    scalar, all sparse indices are set to this single value.
    
    Indices should be sorted in lexicographic order, and indices must not
    contain any repeats. If `validate_indices` is true, these properties
    are checked during execution.
      }];
    
      let arguments = (ins
        TFL_I32OrI64Tensor:$sparse_indices,
        TFL_I32OrI64Tensor:$output_shape,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]
    ```
    
    If `indices` is a permutation and `len(indices) == params.shape[0]` then
    this operation will permute `params` accordingly.
    
    `validate_indices`: DEPRECATED. If this operation is assigned to CPU, values in
    `indices` are always validated to be within range. If assigned to GPU,
    out-of-bound indices result in safe but unspecified behavior, which may include
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top