- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 161 for Indices (0.33 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)