- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CostContext (0.1 sec)
-
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc
} return size; } int64_t InferTensorSize(const CostContext& context, mlir::TensorType type) { if (type.hasRank()) return GetRankedTensorSize(type); return context.default_unranked_tensor_size; } // The cost function for tf.LookupTableFindV2. int64_t InferLookupTableFindV2Cost(const CostContext& context, mlir::TF::LookupTableFindV2Op op) {
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/tfrt/runtime_fallback/runtime_fallback_executor.cc
private: ThreadPool tpool_; }; } // namespace RuntimeFallbackExecutor::RuntimeFallbackExecutor(int64_t num_threads) : intra_op_(std::make_unique<IntraOpThreadPool>(num_threads)) { // Create a HostContext for running TFRT functions. Concurrent work queue acts // similar to the Tensorflow `inter-op` thread pool, so we'll match the size. host_context_ = num_threads ? CreateMultiThreadedHostContext(num_threads)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 9.1K bytes - Viewed (0)