Searched defs:variable_updates (Results 1 – 1 of 1) sorted by relevance
197 const VariableUpdateMap& variable_updates, xla::Backend* backend, in BuildComputationInputs()396 const VariableUpdateMap& variable_updates, TpuNodeContext* node_context, in AllocateOutputTensors()