Searched refs:GetDim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 62 inline int64 GetDim(absl::Span<const int64> data, DimIndex dim) { in GetDim() function
252 int64 height() const { return GetDim(spatial_size(), DimIndex::Y); } in height()
253 int64 width() const { return GetDim(spatial_size(), DimIndex::X); } in width()
254 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim()
434 return GetDim(input_filter_dims(), DimIndex::Y); in input_filter_height()
437 return GetDim(input_filter_dims(), DimIndex::X); in input_filter_width()
440 return GetDim(input_filter_dims(), dim); in input_filter_dim()
562 int64 zero_padding_height() const { return GetDim(padding(), DimIndex::Y); } in zero_padding_height()
563 int64 zero_padding_width() const { return GetDim(padding(), DimIndex::X); } in zero_padding_width()
565 return GetDim(strides(), DimIndex::Y); in vertical_filter_stride()
[all …]
|
| /external/tensorflow/tensorflow/go/ |
| D | signature.go | 111 for _, d := range pb.GetTensorShape().GetDim() {
|