- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 50 for Muto (0.14 sec)
-
tensorflow/c/c_api.cc
CHECK_LT(output.index, ic->num_outputs()); // Crash OK 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());
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
namespace gradients { namespace { class ReluGradientFunction : public GradientFunction { public: explicit ReluGradientFunction(vector<AbstractTensorHandle*> f_outputs) : forward_outputs_(f_outputs) { for (auto output : forward_outputs_) { if (output) { output->Ref(); } } } Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs,
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/eager/c_api_unified_experimental.cc
(GetFactories()[name] == factory) && "Duplicate tracing factory registration"); GetFactories()[name] = factory; } Status SetDefaultTracingEngine(const char* name) { auto entry = GetFactories().find(name); if (entry != GetFactories().end()) { default_factory = GetFactories().find(name)->second; return absl::OkStatus(); } string msg = absl::StrCat(
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/eager/dlpack_test.cc
ASSERT_NE(handle, nullptr) << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",") << "], strides=[" << absl::StrJoin(strides, ",") << "])"; auto* dlm_out = static_cast<DLManagedTensor*>(TFE_HandleToDLPack(handle, status)); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); const DLTensor* dltensor_out = &dlm_out->dl_tensor;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
TF_Tensor* analytical_tensor; auto s = GetValue(t, &analytical_tensor); ASSERT_EQ(errors::OK, s.code()) << s.message(); int64_t num_elem_analytical = 1; auto num_dims_analytical = TF_NumDims(analytical_tensor); ASSERT_EQ(dims.size(), num_dims_analytical); for (int j = 0; j < num_dims_analytical; j++) { auto dim_analytical = TF_Dim(analytical_tensor, j);
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
tensorflow::mutex_lock c(graph->mu); const auto& debug_str = graph->graph.ToGraphDefDebug().DebugString(); *len = debug_str.size(); char* ret = static_cast<char*>(malloc(*len + 1)); memcpy(ret, debug_str.c_str(), *len + 1); return ret; } char* TF_FunctionDebugString(TF_Function* func, size_t* len) { const auto& debug_str = DebugString(func->record->fdef()); *len = debug_str.size();
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/filesystem/plugins/posix/posix_filesystem.cc
} PosixFile; static void Cleanup(TF_WritableFile* file) { auto posix_file = static_cast<PosixFile*>(file->plugin_file); plugin_memory_free(const_cast<char*>(posix_file->filename)); delete posix_file; } static void Append(const TF_WritableFile* file, const char* buffer, size_t n, TF_Status* status) { auto posix_file = static_cast<PosixFile*>(file->plugin_file);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); auto outputs = parallel_device.Execute(context.get(), std::vector<ParallelTensor*>(), "VarHandleOp", TFE_OpGetAttrs(handle_op.get()),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad.cc
absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { for (int i = 0; i < grad_outputs.size(); i++) { auto grad_input = grad_outputs[i]; // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle // that takes care of this similar to `Tensor`? if (grad_input) { grad_input->Ref(); }
C++ - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 1.6K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/custom_gradient_test.cc
ASSERT_EQ(errors::OK, s.code()) << s.message(); TF_Tensor* result_tensor; s = GetValue(outputs[0], &result_tensor); ASSERT_EQ(errors::OK, s.code()) << s.message(); auto result_value = static_cast<float*>(TF_TensorData(result_tensor)); EXPECT_EQ(*result_value, 1.0); outputs[0]->Unref(); TF_DeleteTensor(result_tensor); result_tensor = nullptr; } INSTANTIATE_TEST_SUITE_P(
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0)