- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CostContext (0.09 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) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/BUILD
"@com_google_absl//absl/strings", "@com_google_googletest//:gtest_main", "@tf_runtime//:bef", "@tf_runtime//:befexecutor", "@tf_runtime//:core_runtime", "@tf_runtime//:hostcontext", "@tf_runtime//:init_tfrt_dialects", "@tf_runtime//:support", "@tf_runtime//:tracing", ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/BUILD
"@tf_runtime//:basic_kernels_alwayslink", "@tf_runtime//:bef", "@tf_runtime//:befexecutor", "@tf_runtime//:core_runtime_alwayslink", "@tf_runtime//:hostcontext", "@tf_runtime//:mlirtobef", "@tf_runtime//:support", ], ) cc_library( name = "corert_converter", srcs = [ "transforms/corert_converter.cc", ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 19:04:21 UTC 2024 - 23.6K bytes - Viewed (0)