Searched defs:momentum_buffer (Results 1 – 4 of 4) sorted by relevance
43 explicit SGDParamState(executorch::aten::Tensor& momentum_buffer) in SGDParamState()46 executorch::aten::Tensor& momentum_buffer() { in momentum_buffer() function
188 const at::Tensor& momentum_buffer, in sgd_fused_step_impl()238 const at::Tensor& momentum_buffer, in fused_sgd_kernel()
37 Tensor momentum_buffer = tf.make({2, 2}, {1, 2, 3, 4}); in TEST_F() local
42 const auto momentum_buffer = is_first_step in sgd_math() local