- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 359 for unique_ptr (0.19 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.h
// qint types to int types and surround TF UniformQuantized ops with qint <-> // int casts. std::unique_ptr<OperationPass<func::FuncOp>> CreateConvertTFQuantTypesPass(); // Creates an instance of the VerifyQuantLegalization pass, which verifies all // quant ops and types are lowered. std::unique_ptr<OperationPass<func::FuncOp>> CreateVerifyQuantLegalizationPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 01:41:18 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/jit/clone_constants_for_better_clustering_test.cc
namespace tensorflow { namespace { using ::tensorflow::testing::FindNodeByName; Status CloneConstantsForBetterClustering(const Scope& s, std::unique_ptr<Graph>* result) { auto graph = std::make_unique<Graph>(OpRegistry::Global()); SessionOptions session_options; session_options.config.mutable_graph_options() ->mutable_optimizer_options()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
EXPECT_EQ(0, clusters.size()); } TEST(XlaCompilationTest, DeterministicClusterNames) { auto create_graph = [](absl::string_view output_name) -> std::unique_ptr<Graph> { std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); Tensor t(DT_FLOAT, TensorShape()); t.scalar<float>()() = 0.0f;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_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(); // // XlaHostSendDeviceContext 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.7K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis_test.cc
} return false; } using deadness_analysis_internal::ComputePredicates; using deadness_analysis_internal::PredicateMapTy; Status AnalyzeDeadness(Graph* graph, std::unique_ptr<DeadnessAnalysis>* result) { FixupSourceAndSinkEdges(graph); return DeadnessAnalysis::Run(*graph, result); } ops::Switch CreateSwitch(const Scope& root, const string& prefix) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 51.6K bytes - Viewed (0) -
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/c/eager/parallel_device/parallel_device_lib.cc
class OpDeleter { public: void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); } }; using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>; class StatusDeleter { public: void operator()(TF_Status* to_delete) const { TF_DeleteStatus(to_delete); } }; using StatusPtr = std::unique_ptr<TF_Status, StatusDeleter>; class ExecutorDeleter { public: void operator()(TFE_Executor* to_delete) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 07:47:20 UTC 2024 - 25.4K 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/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/tf2xla/api/v2/tf_executor_to_graph.h
#include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/graph/graph.h" namespace tensorflow { namespace tf2xla { namespace v2 { // Given an MLIR module, returns a GraphDef. absl::StatusOr<std::unique_ptr<GraphDef>> ConvertMlirToGraphdef( mlir::ModuleOp module, const GraphExportConfig& configs); // Converts an MLIR module to TensorFlow graph and FunctionLibraryDefinition.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 3K bytes - Viewed (0)