Searched refs:gradients_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 193 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64> gradients_size, in AvgPoolGrad() argument
202 const int num_gradients = gradients_size.size(); in AvgPoolGrad()
237 out_backprop, gradients_size, kernel_size, stride, spatial_padding, in AvgPoolGrad()
243 std::vector<int64> padded_gradients_size(gradients_size.begin(), in AvgPoolGrad()
244 gradients_size.end()); in AvgPoolGrad()
|
| D | pooling.h | 71 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64> gradients_size,
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 711 std::unordered_map<int64, int64> gradients_size; in ComputeGradient() local
809 auto size_it = gradients_size.find(id); in ComputeGradient()
811 if (size_it == gradients_size.end()) { in ComputeGradient()
813 gradients_size.emplace(id, size); in ComputeGradient()
|