Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 23 for GatherNd (0.3 sec)

  1. tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc

        return false;
      }
    
      // TODO(unda, 253431636): constrained kernel
      std::set<string> constrained_kernel = {"Diag",
                                             "DiagPart",
                                             "GatherNd",
                                             "GatherV2",
                                             "QuantizeAndDequantizeV2",
                                             "QuantizeAndDequantizeV3",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jan 27 16:26:51 UTC 2024
    - 13K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

    // CHECK:           %[[VAL_4:.*]] = "tf.GatherNd"(%[[VAL_3]], %[[VAL_1]]) <{bad_indices_policy = ""}> : {{.*}} -> tensor<4x128xf32>
    // CHECK:           return %[[VAL_4]]
    // CHECK:         }
    // Test the case when start_index_map isn't an iota what requires a transpose to
    // convert it to tf.GatherNd.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

        TF_Int64Tensor:$iter_id
      );
    }
    
    def TF_ResourceGatherNdOp : TF_Op<"ResourceGatherNd", []> {
      let summary = "GatherNd on a resource.";
    
      let description = [{
    This op reads the variable referenced by the first argument, and
    then performs a GatherNd operation on it.
      }];
    
      let arguments = (ins
        Arg<TF_ResourceTensor, "", [TF_VariableRead]>:$resource,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      %0 = "tf.TensorScatterAdd"(%arg0, %arg1, %arg2) : (tensor<7xi64>, tensor<1x1xi32>, tensor<1xi64>) -> tensor<7xi64>
      func.return %0 : tensor<7xi64>
    
      // CHECK-LABEL: scatter_nd_add
      // CHECK:  %[[GATHER:.*]] = "tf.GatherNd"(%arg0, %arg1) <{bad_indices_policy = ""}> : (tensor<7xi64>, tensor<1x1xi32>) -> tensor<1xi64>
      // CHECK:  %[[ADD:.*]] = "tf.Add"(%arg2, %[[GATHER]]) : (tensor<1xi64>, tensor<1xi64>) -> tensor<1xi64>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir

    }
    
    func.func @gatherNdVectorIndices(%arg0 : tensor<3x2x2xf32>, %arg1 : tensor<2xi32>) -> tensor<2xf32> {
      %0 = "tf.GatherNd"(%arg0, %arg1) : (tensor<3x2x2xf32>, tensor<2xi32>) -> tensor<2xf32>
      func.return %0 : tensor<2xf32>
    
    // CHECK-LABEL:gatherNdVectorIndices
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 05 01:54:33 UTC 2024
    - 153.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      func.return %0: tensor<4x?xf32>
    }
    
    //===----------------------------------------------------------------------===//
    // tf.GatherNd legalization
    //===----------------------------------------------------------------------===//
    // CHECK-LABEL: func @gatherNd_dynamic
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  8. maven-model-builder/src/main/java/org/apache/maven/model/building/DefaultTransformerContextBuilder.java

                            return defaultModelBuilder.readRawModel(gaBuildingRequest, problems);
                        } catch (ModelBuildingException e) {
                            // gathered with problem collector
                        }
                    }
                    return null;
                }
    
                private void loadFullReactor() {
                    if (!fullReactorLoaded) {
    Registered: Wed Jun 12 09:55:16 UTC 2024
    - Last Modified: Mon Feb 26 17:04:44 UTC 2024
    - 8.8K bytes
    - Viewed (0)
  9. maven-api-impl/src/main/java/org/apache/maven/internal/impl/model/DefaultModelTransformerContextBuilder.java

                            return defaultModelBuilder.readRawModel(gaBuildingRequest, problems);
                        } catch (ModelBuilderException e) {
                            // gathered with problem collector
                        }
                    }
                    return null;
                }
    
                private void loadFullReactor() {
                    if (!fullReactorLoaded) {
    Registered: Wed Jun 12 09:55:16 UTC 2024
    - Last Modified: Fri Apr 12 10:50:18 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  10. RELEASE.md

    *   Fixes a division by zero in TFLite's implementation of `SpaceToDepth`
        ([CVE-2021-29587](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2021-29587))
    *   Fixes a division by zero in TFLite's implementation of `GatherNd`
        ([CVE-2021-29589](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2021-29589))
    *   Fixes a division by zero in TFLite's implementation of `TransposeConv`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 730.3K bytes
    - Viewed (0)
Back to top