- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for Op (0.18 sec)
-
tensorflow/c/c_api_test.cc
TF_DeleteStatus(status); ASSERT_EQ(TF_OK, code) << status_msg; // Test op list. TF_Buffer op_list_buf = TF_GetOpList(lib); tensorflow::OpList op_list; EXPECT_TRUE(op_list.ParseFromArray(op_list_buf.data, op_list_buf.length)); ASSERT_EQ(op_list.op_size(), 1); EXPECT_EQ("TestCApi1", op_list.op(0).name()); TF_DeleteLibraryHandle(lib); } #endif // !defined(TENSORFLOW_NO_SHARED_OBJECTS)
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/eager/unified_api_testutil.h
absl::Span<AbstractTensorHandle* const>, absl::Span<AbstractTensorHandle*>)>; // Runs `model` maybe wrapped in a function call op. This can be thought as // being equivalent to the following python code. // // if use_function: // outputs = tf.function(model)(inputs) // else: // outputs = model(inputs)
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
GradientFunction* ReluRegisterer(const ForwardOperation& op) { return new ReluGradientFunction(op.outputs); } GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer( const ForwardOperation& op) { return new SparseSoftmaxCrossEntropyWithLogitsGradientFunction(op.outputs); } GradientFunction* BiasAddRegisterer(const ForwardOperation& op) { return new BiasAddGradientFunction(op.attrs); } } // namespace gradients
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.cc
delete lib_handle; } TF_Buffer* TF_GetAllOpList() { std::vector<tensorflow::OpDef> op_defs; tensorflow::OpRegistry::Global()->GetRegisteredOps(&op_defs); tensorflow::OpList op_list; for (const auto& op : op_defs) { *(op_list.add_op()) = op; } TF_Buffer* ret = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(op_list, ret)); return ret; }
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/eager/c_api.cc
void TFE_DeleteOp(TFE_Op* op) { if (op == nullptr) { return; } tensorflow::unwrap(op)->Release(); } const char* TFE_OpGetName(const TFE_Op* op, TF_Status* status) { return tensorflow::unwrap(op)->Name().c_str(); } TFE_Context* TFE_OpGetContext(const TFE_Op* op, TF_Status* status) { return tensorflow::wrap(tensorflow::unwrap(op)->GetContext()); }
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/parallel_device/BUILD
"//tensorflow/c/eager:c_api", "//tensorflow/c/eager:c_api_experimental", "//tensorflow/c/eager:tfe_cancellation_manager_internal", "//tensorflow/c/eager:tfe_op_internal", "//tensorflow/c/eager:tfe_tensorhandle_internal", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/types:optional",
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 01 20:19:06 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
Status Reset(const char* op, const char* raw_device_name) override { if (op_) { return errors::FailedPrecondition("Reset called on already built op."); } if (raw_device_name) { device_name_ = raw_device_name; } op_type_ = op; return absl::OkStatus(); } Status SetOpName(const char* const op_name) override { if (op_) { return errors::FailedPrecondition(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/tracing_utils.cc
namespace tracing { Status MaybeSetOpName(AbstractOperation* op, const char* op_name) { if (isa<TracingOperation>(op)) { TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name)); } if (isa<gradients::TapeOperation>(op)) { TF_RETURN_IF_ERROR(MaybeSetOpName( dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(), op_name)); } return absl::OkStatus(); } } // namespace tracing
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad.cc
} // namespace GradientFunction* AddRegisterer(const ForwardOperation& op) { return new AddGradientFunction; } GradientFunction* ExpRegisterer(const ForwardOperation& op) { return new ExpGradientFunction(op.outputs[0]); } GradientFunction* MatMulRegisterer(const ForwardOperation& op) { return new MatMulGradientFunction(op.inputs, op.attrs); } GradientFunction* SqrtRegisterer(const ForwardOperation& op) {
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
auto op_id_it = tensor_tape.find(tensor_id); if (op_id_it == tensor_tape.end()) { continue; } int64_t op_id = op_id_it->second; auto op_it = op_tape->find(op_id); auto result_op_it = result.op_tape.find(op_id); if (op_id == -1 || op_it == op_tape->end() || result_op_it != result.op_tape.end()) { continue; } CHECK(result.op_tape.emplace(op_id, op_it->second).second);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1)