Home
last modified time | relevance | path

Searched defs:tape (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dtape.py36 def __init__(self, tape): argument
49 def push_tape(tape): argument
54 def watch(tape, tensor): argument
91 def watch_variable(tape, variable): argument
145 def pop_tape(tape): argument
H A Dpywrap_tfe_src.cc1617 GradientTape* tape; member
1624 static void TFE_Py_Tape_Delete(PyObject* tape) { in TFE_Py_Tape_Delete()
1930 TFE_Py_Tape* tape = PyObject_NEW(TFE_Py_Tape, &TFE_Py_Tape_Type); in TFE_Py_TapeSetNew() local
1939 void TFE_Py_TapeSetAdd(PyObject* tape) { in TFE_Py_TapeSetAdd()
1957 void TFE_Py_TapeSetRemove(PyObject* tape) { in TFE_Py_TapeSetRemove()
2122 void TFE_Py_TapeWatch(PyObject* tape, PyObject* tensor) {
2276 void TFE_Py_TapeWatchVariable(PyObject* tape, PyObject* variable) {
2283 PyObject* TFE_Py_TapeWatchedVariables(PyObject* tape) {
2796 PyObject* TFE_Py_TapeGradient(PyObject* tape, PyObject* target,
H A Dimperative_grad.py29 def imperative_grad(tape, argument
/aosp_15_r20/external/clang/test/SemaCXX/
H A Dconstexpr-turing.cpp10 bool tape; member
35 constexpr unsigned run(const State *tm, const Tape &tape, unsigned state) { in run()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dgradients_test.cc106 auto tape = std::make_unique<Tape>(/*persistent=*/false); in TEST_P() local
120 Tape tape(/*persistent=*/false); in RecordOperationWithNullGradientFunctionModel() local
H A Dgradients.cc469 ForwardOperation* forward_op_, Tape* tape, in Execute()
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc1045 [](const py::handle& tape) { TFE_Py_TapeSetAdd(tape.ptr()); }); in PYBIND11_MODULE()
1047 [](const py::handle& tape) { TFE_Py_TapeSetRemove(tape.ptr()); }); in PYBIND11_MODULE()
1106 [](const py::handle& tape, const py::handle& tensor) { in PYBIND11_MODULE()
1110 [](const py::handle& tape, const py::handle& variable) { in PYBIND11_MODULE()
1113 m.def("TFE_Py_TapeWatchedVariables", [](const py::handle& tape) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/
H A Doptimizer_v2.py465 def _get_gradients(self, tape, loss, var_list, grad_loss=None): argument
502 def minimize(self, loss, var_list, grad_loss=None, name=None, tape=None): argument
537 def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/tape/
H A Dtape_context.cc22 TapeContext::TapeContext(AbstractContext* c, Tape* tape, in TapeContext()
H A Dtape_operation.cc22 TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape, in TapeOperation()
/aosp_15_r20/external/pigweed/pw_bluetooth_sapphire/fuchsia/host/fidl/measure_tape/
H A Dhlcpp_measure_tape_for_read_by_type_result.cc103 MeasuringTape tape; in Measure() local
H A Dhlcpp_measure_tape_for_peer.cc221 MeasuringTape tape; in Measure() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dtape.cc70 [](AbstractContext* ctx, Tape* tape, GradientRegistry* registry) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/
H A Dcustom_gradient_test.cc71 Tape tape(/*persistent=*/false); in ExpWithPassThroughGrad() local
H A Dgrad_test_helper.cc115 Tape tape(/*persistent=*/false); in BuildGradModel() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Doptimizer_v1.py782 def minimize(self, loss, var_list, grad_loss=None, tape=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dloss_scale_optimizer.py569 def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): argument
/aosp_15_r20/external/sg3_utils/src/
H A Dsg_format.c102 int tape; /* -T <format>, def: -1 */ member
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/
H A Dtokenizer.json67032 "tape": 64620, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/
H A Dtokenizer.json67032 "tape": 64620, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/
H A Dtokenizer.json66973 "tape": 64620, number