- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for CostContext (0.09 sec)
-
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.h
int64_t max_arg_size_ = 1; llvm::DenseMap<mlir::Operation*, int64_t> cost_map_; const tfrt_stub::CostRecorder* cost_recorder_; }; struct CostContext { int64_t default_unranked_tensor_size; }; using CostFunction = std::function<int64_t(const CostContext&, mlir::Operation*)>; void RegisterCostFunction(absl::string_view op_name, CostFunction cost_function);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 3.1K bytes - Viewed (0) -
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/tests/tfrt_fallback/batch_function_fallback_benchmark_test.cc
// Creates a BEF file with a program that runs // tfrt_fallback_async.batch_function with a empty function forwarding inputs or // outputs. std::pair<tfrt::BefBuffer, tfrt::RCReference<tfrt::BEFFile>> CreateBefFile( tfrt::HostContext* host) { std::string file_path = GetDataDependencyFilepath( "tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/" "batch_function_fallback.mlir.bef"); std::string data;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 08 08:08:48 UTC 2023 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.h
llvm::ArrayRef<Tensor> arguments); private: void RunTfrtInitializer(); std::unique_ptr<thread::ThreadPoolInterface> intra_op_; std::unique_ptr<tfrt::HostContext> host_context_; tfrt::ResourceContext resource_context_; std::unique_ptr<tfrt::ExecutionContext> exec_ctx_; tfrt::BefBuffer bef_buffer_; tfrt::RCReference<tfrt::BEFFile> bef_file_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 14 22:40:40 UTC 2022 - 2.4K 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)