Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 161 for Indices (0.49 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      let description = [{
    This operation computes
    
        # Scalar indices
        ref[indices, ...] = max(ref[indices, ...], updates[...])
    
        # Vector indices (for each i)
        ref[indices[i], ...] = max(ref[indices[i], ...], updates[i, ...])
    
        # High rank indices (for each i, ..., j)
        ref[indices[i, ..., j], ...] = max(ref[indices[i, ..., j], ...], updates[i, ..., j, ...])
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/decompose_resource_op.mlir

    // CHECK-NEXT: tfrt.return [[out_chain]], [[res]] : !tfrt.chain, !tfrt_fallback.tf_tensor
    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>)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 00:40:32 UTC 2024
    - 1.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc

        indices = rewriter.create<mhlo::ReshapeOp>(parent_op->getLoc(),
                                                   indices_type, indices);
      } else if (index_vector_dim != indices_type.getRank() - 1) {
        // If index_vector_dim isn't the last dimension in indices then it isn't
        // supported yet.
        // TODO(tberghammer): Transpose indices to support this usecase.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

    def LegalizeGather: Pat<
      (TF_GatherOp $params, $indices, $ignored_validate_indices),
      (TFL_GatherOp $params, $indices, ConstantAttr<I32Attr, "0">,
         ConstantAttr<I32Attr, "0">)>;
    
    def LegalizeGatherNd : Pat<(TF_GatherNdOp $params, $indices, $bad_indices_policy),
                               (TFL_GatherNdOp $params, $indices)>;
    
    def LegalizeGatherV2 : Pat<
      (TF_GatherV2Op $params, $indices, (Arith_ConstantOp ElementsAttr:$axis), $batch_dims),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  5. src/go/types/check_test.go

    					continue
    				}
    			}
    			indices = append(indices, i)
    		}
    		if len(indices) == 0 {
    			t.Errorf("%s: no error expected: %q", gotPos, gotMsg)
    			continue
    		}
    		// len(indices) > 0
    
    		// If there are multiple matching errors, select the one with the closest column position.
    		index := -1 // index of matching error
    		var delta int
    		for _, i := range indices {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 22 19:45:33 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.cc

        }
      }
      return resource_vec;
    }
    
    LogicalResult _TfrtGetResourceOp::verify() {
      _TfrtGetResourceOp get_resource_op = *this;
      // The sizes of indices, shared_name and container must be equal.
      int32_t indices_size =
          get_resource_op->getAttrOfType<mlir::ArrayAttr>("indices").size();
      int32_t shared_name_size =
          get_resource_op->getAttrOfType<mlir::ArrayAttr>("shared_name").size();
      int32_t container_size =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.6K bytes
    - Viewed (0)
  7. src/cmd/compile/internal/types2/check_test.go

    					continue
    				}
    			}
    			indices = append(indices, i)
    		}
    		if len(indices) == 0 {
    			t.Errorf("%s: no error expected: %q", gotPos, gotMsg)
    			continue
    		}
    		// len(indices) > 0
    
    		// If there are multiple matching errors, select the one with the closest column position.
    		index := -1 // index of matching error
    		var delta uint
    		for _, i := range indices {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 22 19:45:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td

    def DecomposeResourceScatterAdd : Pat<
      (TF_ResourceScatterAddOp:$src_op $resource, $indices, $updates),
      (TF_AssignVariableOp
        $resource,
        (TF_TensorScatterAddOp
          (CreateTFReadVariableOp $src_op, $updates, $resource),
          (TF_ExpandDimsOp $indices,
           (TF_ConstOp (GetScalarOfType<-1> $indices))),
          $updates
        ),
        (CreateConstBoolAttrFalse)
      )>;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 20.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.cc

      if (!ranked_ty) return UnrankedTensorType::get(element_ty);
      int64_t rank = ranked_ty.getRank();
    
      DenseIntElementsAttr indices;
      if (!matchPattern(reduction_indices, m_Constant(&indices))) {
        // Output type is unranked if reduction indices are not constant and reduced
        // dimensions are not kept.
        if (!keep_dims.getValue()) return UnrankedTensorType::get(element_ty);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

        return success();
      }
    };
    
    // Lowers DynamicStitch op with constant indices and with static input and
    // output shapes using Reshape, UnPack and Pack op.
    //
    //   %indices0 = "tf.Const"() {value = dense<4> : tensor<i32>}
    //   %indices1 = "tf.Const"() {value = dense<[[3, 2], [1, 0]]> :
    //   tensor<2x2xi32>} %0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0,
    //   %arg1)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
Back to top