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