Searched refs:tensor_array_gradients_ (Results 1 – 2 of 2) sorted by relevance
82 tensor_array_gradients_[gradient].reset(new XlaResource( in XlaResource()171 std::unique_ptr<XlaResource>& gradient = tensor_array_gradients_[source]; in GetOrCreateTensorArrayGradient()190 if (tensor_array_gradients_.empty()) { in Pack()196 for (const auto& gradient : tensor_array_gradients_) { in Pack()
166 return tensor_array_gradients_; in tensor_array_gradients()186 std::map<string, std::unique_ptr<XlaResource>> tensor_array_gradients_; variable