- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 34 for shared_ptr (0.94 sec)
-
tensorflow/compiler/jit/xla_platform_info.cc
se::Platform::Id platform_id = nullptr; const XlaDevice::Metadata* xla_device_metadata = nullptr; const PjRtBaseDevice::Metadata* pjrt_device_metadata = nullptr; std::shared_ptr<se::DeviceMemoryAllocator> custom_allocator; const std::string& device_type = device_base->device_type(); if (device_type == DEVICE_CPU) { platform_id = se::host::kHostPlatformId;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator_test.cc
#include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/platform/test.h" #include "tensorflow/core/public/session_options.h" #include "tensorflow/core/public/version.h" namespace tensorflow { std::shared_ptr<NodeProperties> ToNodeProperties(const string& text) { NodeDef node_def; DataTypeVector dummy; EXPECT_TRUE(protobuf::TextFormat::MergeFromString(text, &node_def));
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/jit/compilability_check_util.h
public: // Contains node name and function name. If the node is not inside a function // body, function name is an empty string. struct StackFrame { std::string name; std::string function_name; std::shared_ptr<AbstractStackTrace> stack_trace; }; // Contains information about uncompilable node inside a function body. struct UncompilableNodeInfo { std::string name;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
DataType output_dtype, const TensorShape& output_shape, Allocator* output_allocator, bool allocate_xla_tensors, se::Stream* stream, bool use_multiple_streams, std::shared_ptr<se::Event> definition_event) { xla::ShapeIndex output_index = input_output_alias.shape().IsTuple() ? xla::ShapeIndex({output_num})
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
xla::LocalClient* client = static_cast<xla::LocalClient*>(xla_device_compiler->client()); se::Stream* stream = ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; std::shared_ptr<se::DeviceMemoryAllocator> allocator_ptr = GetAllocator(ctx->device(), stream, platform_info_); se::DeviceMemoryAllocator* allocator = allocator_ptr.get(); XlaComputationLaunchContext launch_context(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/cc/framework/scope.h
bool ok() const; // TODO(skyewm): Graph is not part of public API Graph* graph() const; // TODO(skyewm): Graph is not part of public API std::shared_ptr<Graph> graph_as_shared_ptr() const; Status status() const; /// If status() is ok, convert the Graph object stored in this scope /// to a GraphDef proto and return an ok Status. Otherwise, return the error
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 09:08:33 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
} return module_op; } // Evaluate return value of a function using TF kernel. // This assumes that the module op has only 1 function and it has TF ops only. absl::StatusOr<std::shared_ptr<xla::Literal>> EvaluateTfFunction( absl::string_view program, absl::Span<const xla::Literal* const> arguments) { TF_ASSIGN_OR_RETURN(auto module_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
std::map<int, const Tensor*> resource_var_ptrs; for (int i = 0; i < resources.size(); i++) { resource_var_ptrs[resources[i]] = variable_infos[i].var()->tensor(); } std::shared_ptr<se::DeviceMemoryAllocator> allocator = GetAllocator(ctx->device(), GetStream(ctx), platform_info); XlaComputationLaunchContext launch_context =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
// triggered. TF_RETURN_IF_ERROR(dst_device_to_device_stream->WaitFor( dst_xla_context->host_to_device_stream())); } TF_ASSIGN_OR_RETURN(std::shared_ptr<se::Event> definition_event, dst_xla_context->stream()->parent()->CreateEvent()); TF_RETURN_IF_ERROR( dst_device_to_device_stream->RecordEvent(definition_event.get()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
return op_->emitRemark() << "failed to convert op to NodeDef: " << nodedef_or.status().ToString(); } if (failed(PrepareParams())) return failure(); std::shared_ptr<const tensorflow::NodeProperties> props; absl::Status status = tensorflow::NodeProperties::CreateFromNodeDef( *nodedef_or.value(), params_.function_library->GetFunctionLibraryDefinition(), &props);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0)