- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 29 for types (0.17 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
#include "tensorflow/c/tf_datatype.h" #include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status_helper.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h" #include "tensorflow/core/platform/errors.h" #include "tensorflow/core/platform/types.h" using tensorflow::string; namespace tensorflow { namespace tracing {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/c_api.cc
const auto* shapes_and_types = ic->output_handle_shapes_and_types(output.index); if (shapes_and_types == nullptr) return nullptr; for (const auto& p : *shapes_and_types) { auto* out_shape_and_type = handle_data.add_shape_and_type(); ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape()); out_shape_and_type->set_dtype(p.dtype);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/experimental/gradients/nn_grad.h" #include "absl/types/span.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/eager/immediate_execution_context.h" #include "tensorflow/c/eager/immediate_execution_tensor_handle.h"
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
TF_ShapeAndTypeList* input_tensors_as_shapes, TF_ShapeAndTypeList** input_resource_shapes_and_types, TF_ShapeAndTypeList** output_shapes, TF_ShapeAndTypeList*** output_resource_shapes_and_types, TF_Status* status) { using tensorflow::NodeDef; using tensorflow::OpRegistrationData; using tensorflow::Tensor;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
tensorflow::DeviceType(device_type), std::move(*pjrt_client)); status->status = s; } void TF_ResetPjRtCClient(const char* device_type, TF_Status* status) { status->status = tensorflow::ResetPjRtClient(tensorflow::DeviceType(device_type)); } PJRT_Client* TF_GetPjRtCClient(const char* device_type, TF_Status* status) { absl::StatusOr<xla::PjRtCApiClient*> pjrt_c_api_client =
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
#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" #include "tensorflow/core/tfrt/common/async_value_tensor.h" #include "tensorflow/core/tfrt/common/pjrt_util.h" #include "tsl/lib/core/status_test_util.h" #include "tsl/platform/casts.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/eager/parallel_device/parallel_device.cc
"The allocated output buffer was too small."); return; } for (int i = 0; i < typed_outputs.size(); ++i) { MaybeParallelTensorOwned typed_output(std::move(typed_outputs[i])); if (absl::holds_alternative<TensorHandlePtr>(typed_output)) { outputs[i] = absl::get<TensorHandlePtr>(typed_output).release(); } else { outputs[i] = ParallelTensorToTensorHandle(
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/gradient_checker.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/gradient_checker.h" #include <memory> #include "absl/types/span.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/experimental/ops/math_ops.h" #include "tensorflow/c/tf_tensor.h" namespace tensorflow { namespace gradients {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
==============================================================================*/ #include "tensorflow/c/eager/gradients.h" #include <memory> #include "absl/container/flat_hash_set.h" #include "absl/types/span.h" #include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/c/eager/c_api_test_util.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
// Pack 3 variable handles into one TFE_TensorHandle. // When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1. int num_replicas = 3; std::vector<TFE_TensorHandle*> handles = {h0, h1, h2}; TFE_TensorHandle* packed_handle =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0)