Searched refs:gradients_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 192 XlaOp AvgPoolGrad(XlaOp out_backprop, absl::Span<const int64> gradients_size, in AvgPoolGrad() argument
202 if (gradients_size.size() != num_dims) { in AvgPoolGrad()
234 out_backprop, gradients_size, kernel_size, stride, spatial_padding, in AvgPoolGrad()
240 std::vector<int64> padded_gradients_size(gradients_size.begin(), in AvgPoolGrad()
241 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 | 504 gtl::FlatMap<int64, int64> gradients_size; in ComputeGradient() local
588 auto size_it = gradients_size.find(id); in ComputeGradient()
590 if (size_it == gradients_size.end()) { in ComputeGradient()
592 gradients_size.emplace(id, size); in ComputeGradient()
|