- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GatherV2 (0.13 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/jit/tests/opens2s_gnmt_mixed_precision.golden_summary
ExpandDims 1 ReverseSequence 1 Shape 1 StridedSlice 1 Transpose 3 cluster 6 size 11 Cast 1 Const 5 GatherV2 1 Shape 1 StridedSlice 1 Transpose 1 ZerosLike 1 cluster 7 size 33 All 2 Cast 1 Const 17 Equal 2 ExpandDims 2 GatherV2 1 Identity 1 ReverseSequence 1 Shape 2 StridedSlice 2 Transpose 2 cluster 8 size 11 Cast 1 Const 4
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 06 10:38:14 UTC 2023 - 5K 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/quantization/tensorflow/passes/quantized_function_library_xla_weight_only.mlir
%weight_scale : tensor<*xf32>, %weight_zp : tensor<*xi32>) -> tensor<*xf32> attributes {tf_quant.quantized_ops = ${quantized_ops}} { %accum_out = "tf.GatherV2"(%weight, %input, %axis) { batch_dims = 0 : i64, attr_map = "batch_dims:0"} : (tensor<*xi8>, tensor<*xi32>, tensor<i32>) -> tensor<*xi8> %accum_out_new = "tf.Cast"(%accum_out) : (tensor<*xi8>) -> tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 15:43:38 UTC 2023 - 7K 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)