Searched refs:GetDim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 61 inline int64 GetDim(absl::Span<const int64> data, DimIndex dim) { in GetDim() function
260 int64 height() const { return GetDim(spatial_size(), DimIndex::Y); }
261 int64 width() const { return GetDim(spatial_size(), DimIndex::X); }
262 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); }
442 return GetDim(input_filter_dims(), DimIndex::Y);
445 return GetDim(input_filter_dims(), DimIndex::X);
448 return GetDim(input_filter_dims(), dim);
566 int64 zero_padding_height() const { return GetDim(padding(), DimIndex::Y); }
567 int64 zero_padding_width() const { return GetDim(padding(), DimIndex::X); }
569 return GetDim(strides(), DimIndex::Y);
[all …]
|