- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 641 for unique_ptr (0.13 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.h
absl::StatusOr<std::string> ConvertSerializedStableHloModuleToBfloat16( StringRef serialized_stablehlo_module); std::unique_ptr<OperationPass<ModuleOp>> CreateLiftQuantizableSpotsAsFunctionsPass( const ::stablehlo::quantization::QuantizationSpecs& quantization_specs); // Creates a pass that inserts CalibrationStatisticsSaverOp. std::unique_ptr<OperationPass<ModuleOp>> CreateInsertCalibrationStatisticsSaverPass( StringRef calibration_data_dir,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 2.6K 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/jit/pjrt_tensor_buffer_util.cc
return shape.num_elements() * DataTypeSize(dtype); } absl::StatusOr<Tensor> MakeTensorFromPjRtBuffer( const DataType dtype, const TensorShape& shape, std::unique_ptr<xla::PjRtBuffer> pjrt_buffer) { TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::PjRtBuffer::ExternalReference> ref, pjrt_buffer->AcquireExternalReference()); auto* tensor_buffer = new PjRtTensorBuffer(ref->OpaqueDeviceMemoryDataPointer(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 14 18:14:47 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_variable_loading_test.cc
std::vector<std::unique_ptr<Device>> devices; TF_CHECK_OK(DeviceFactory::AddDevices( options, "/job:localhost/replica:0/task:0", &devices)); device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); ctx_ = testing::CreateTestingEagerContext(device_mgr_.get()); } EagerContext* context() { return ctx_.get(); } private: std::unique_ptr<StaticDeviceMgr> device_mgr_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 09:34:33 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_test.cc
EXPECT_EQ(nodes_preserved.size(), num_values); EXPECT_EQ(list_total_size, storage_size); std::unique_ptr<char*[]> values(new char*[nodes_preserved.size()]); std::unique_ptr<size_t[]> lens(new size_t[nodes_preserved.size()]); std::unique_ptr<char[]> storage(new char[storage_size]); TF_GetNodesToPreserveList(c_item, values.get(), lens.get(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 11.6K bytes - Viewed (0) -
tensorflow/cc/saved_model/experimental/tests/saved_model_api_test.cc
} builder.SetUseTFRT(use_tfrt); std::unique_ptr<Runtime> runtime = builder.Build(&status); ASSERT_TRUE(status.ok()) << status.message(); std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); std::unordered_set<std::string> tags = {"serve"}; std::unique_ptr<SavedModelAPI> model = SavedModelAPI::Load(model_dir, *runtime, &status, &tags);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 14 23:59:14 UTC 2020 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.h
// se::DeviceMemoryBase gpu_dst{device_tensor.data(), 4 * sizeof(float)}; // xla::Shape shape(xla::F32, {2, 2}, {}, {}) // tsl::AsyncValueRef<std::unique_ptr<se::Event>> done_event = // tsl::MakeConstructedAsyncValueRef<std::unique_ptr<se::Event>>(stream.parent()); // done_event->Init(); // Tensor dest_cpu_tensor; // // XlaHostRecvDeviceContext device_context(&stream, gpu_dst, // shape, done_event);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.h
public: explicit MlirOptimizationPassRegistration( int priority, std::unique_ptr<MlirOptimizationPass> pass) { MlirOptimizationPassRegistry::Global().Add(priority, std::move(pass)); } }; class MlirV1CompatOptimizationPassRegistration { public: explicit MlirV1CompatOptimizationPassRegistration( std::unique_ptr<MlirV1CompatOptimizationPass> pass) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 03 22:53:34 UTC 2023 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator_test.cc
flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); } FunctionLibraryRuntime* flr_; std::unique_ptr<DeviceMgr> device_mgr_; std::unique_ptr<FunctionLibraryDefinition> lib_def_; std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; std::unique_ptr<OpKernel> kernel_; }; AttrValue BoolAttr(bool b) { AttrValue v; v.set_b(b); return v; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/debug/debug.cc
file = nullptr; } } // The file being written to. std::unique_ptr<tsl::WritableFile> file; }; // Reproducer stream that emits a reproducer to the given `llvm::raw_ostream`. class ReproducerStream : public mlir::ReproducerStream { public: ReproducerStream(std::string name, std::unique_ptr<llvm::raw_ostream> os) : name_(std::move(name)), os_(std::move(os)) {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 10 02:44:52 UTC 2024 - 11.8K bytes - Viewed (0)