- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for InferGatherV2Cost (0.46 sec)
-
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())); } // The cost function for tf.SparseSegmentSumOp.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0)