Searched refs:tensor_array_gradients_ (Results 1 – 2 of 2) sorted by relevance
80 tensor_array_gradients_[gradient].reset(new XlaResource( in XlaResource()168 std::unique_ptr<XlaResource>& gradient = tensor_array_gradients_[source]; in GetOrCreateTensorArrayGradient()187 if (tensor_array_gradients_.empty()) { in Pack()193 for (const auto& gradient : tensor_array_gradients_) { in Pack()
163 return tensor_array_gradients_; in tensor_array_gradients()183 std::map<string, std::unique_ptr<XlaResource>> tensor_array_gradients_; variable