- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for GatherV2 (0.17 sec)
-
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/decompose_resource_op.mlir
// CHECK-NEXT: [[res:%.*]] = tfrt_fallback_async.executeop key(1) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.GatherV2"([[value]], {{.*}}, [[const]]) // CHECK-NEXT: tfrt.return [[out_chain]], [[res]] : !tfrt.chain, !tfrt_fallback.tf_tensor func.func @gather(%indices: 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/tfrt/tests/mlrt/while_to_map_fn.mlir
// CHECK-NEXT: tf.Const // CHECK-NEXT: tf.Const // CHECK-NEXT: tf.Const // CHECK-NEXT: tf.Const // CHECK-NEXT: tf.GatherV2 // CHECK-NEXT: tf.AddV2 // CHECK-NEXT: tf.AddV2 // CHECK-NEXT: tf.GatherV2 // CHECK-NEXT: tf.Mul // CHECK-NEXT: tf.GatherV2 // CHECK-NEXT: tf.Mul // CHECK-NEXT: tf_mlrt.tf_await // CHECK-NEXT: tf_mlrt.tf_await // CHECK-NEXT: tf.TensorListSetItem
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:40:22 UTC 2024 - 68.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir
} } // ----- // Decomposable Resource Ops usage // // CHECK-LABEL: func.func @serving_default // CHECK: "tf.VarHandleOp" // CHECK-NEXT: "tf.IfrtLoadVariable" // CHECK-NEXT: "tf.GatherV2" // CHECK-NEXT: return // module { func.func @serving_default() -> tensor<1x3xbf16> { %cst = "tf.Const"() <{value = dense<[1]> : tensor<1xi32>}> : () -> tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 15:33:17 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/fuse-tftext.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 460.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc
if (mlir::isa<mlir::TF::StringType>(key_type.getElementType())) cost *= kLookupTableFindStringKeyCostScale; return cost; } // The cost function for tf.GatherV2. int64_t InferGatherV2Cost(const CostContext& context, mlir::TF::GatherV2Op op) { return InferTensorSize( context, mlir::cast<mlir::TensorType>(op.getOutput().getType())); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td
(TF_AssignVariableOp $v_resource, $new_v, (CreateConstBoolAttrFalse)) ] >; // 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),
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/tests/decompose_resource_ops.mlir
%resource = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<*xi32>>> // CHECK-DAG: [[READVAR:%.+]] = "tf.ReadVariableOp"([[VAR]]) // CHECK: [[GATHER:%.+]] = "tf.GatherV2"([[READVAR]], [[INDEX]], [[ZERO]]) <{batch_dims = 0 : i64}> {_xla_outside_compilation = "0"} : (tensor<*xi32>, tensor<?xi32>, tensor<i64>) -> tensor<*xi32> // CHECK: return [[GATHER]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
(MHLO_FftOp $input, MHLO_FftTypeValue<"IFFT">, (GetInnerDimFromValue $res)), [(CheckInnerDimStatic $input)]>; //===----------------------------------------------------------------------===// // GatherV2 op patterns. //===----------------------------------------------------------------------===// // Here, $params and $indices needs to be ranked so that $axis and $batch_dims
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
// CHECK: %[[CST:.*]] = "tf.Const"() <{value = dense<0> : tensor<i32>}> : () -> tensor<i32> // CHECK: %[[RES0:.*]] = "tf.GatherV2"(%arg0, %arg2, %cst) <{batch_dims = 0 : i64}> {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<1600x1x32xf32>, tensor<i32>, tensor<i32>) -> tensor<1x32xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0)