- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for Wrapper (2.21 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
#include <optional> #include <utility> #include <vector> #include <gtest/gtest.h> #include "absl/log/check.h" #include "xla/pjrt/c/pjrt_c_api.h" #include "xla/pjrt/c/pjrt_c_api_cpu.h" #include "xla/pjrt/c/pjrt_c_api_wrapper_impl.h" #include "xla/pjrt/pjrt_api.h" #include "xla/pjrt/pjrt_c_api_client.h" #include "xla/pjrt/tfrt_cpu_pjrt_client.h" #include "xla/shape.h" #include "xla/shape_util.h" #include "tensorflow/core/framework/types.h"
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler.cc
#include "tensorflow/core/grappler/grappler_item.h" #include "tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.h" #include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/status.h" #include "tensorflow/core/protobuf/rewriter_config.pb.h" #include "tsl/platform/env.h" #include "tsl/platform/errors.h" namespace {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1) -
tensorflow/c/experimental/grappler/grappler_test.cc
#include "tensorflow/c/experimental/grappler/grappler.h" #include "tensorflow/c/c_api_internal.h" #include "tensorflow/c/experimental/grappler/grappler_internal.h" #include "tensorflow/c/tf_buffer_internal.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/grappler/clusters/single_machine.h" #include "tensorflow/core/grappler/costs/graph_properties.h"
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
} result.emplace(std::move(result_content)); return result; } // Used as an argument to TFE_NewCustomDeviceTensorHandle, indicating how // ParallelTensors wrapped in TFE_TensorHandles should be cleaned up once their // reference counts drop to zero. void ParallelTensorDeallocator(void* data) { delete reinterpret_cast<ParallelTensor*>(data); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(ctx)->AddParameter( input->DataType(), shape, &handle)); params->emplace_back(handle); } return absl::OkStatus(); } // Runs `model` maybe wrapped in a function. Status RunModel(Model model, AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs, bool use_function) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} // Set server_def on the context, possibly updating it. // TODO(b/291142876) Simplify TFE_ContextSetServerDefWithTimeoutAndRetries and // TFE_ContextUpdateServerDefWithTimeout to be simple wrappers around the same // C++ function. // Retries are used for CreateContext calls, which is used in // ParameterServerStrategy initialization to be robust to worker preemption.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // Add a config_proto attr, to trigger grappler graph rewrites in the current // eager runtime. if (enable_grappler) { tensorflow::ConfigProto config; // Do not skip grappler optimization even for small graphs. config.mutable_graph_options() ->mutable_rewrite_options() ->set_min_graph_nodes(-1);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1)