Searched refs:GetFilterDim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 446 T GetFilterDim(gtl::ArraySlice<T> dimension_attribute, in GetFilterDim() function
476 inline int64 GetFilterDim(const TensorShape& tensor_shape, in GetFilterDim() function
479 return GetFilterDim(gtl::ArraySlice<int64>(tensor_shape.dim_sizes()), in GetFilterDim()
492 inline int64 GetFilterDim(const Tensor& tensor, in GetFilterDim() function
495 return GetFilterDim(tensor.shape(), filter_tensor_format, dimension); in GetFilterDim()
623 const int64 output_channels = GetFilterDim(src_shape, src_filter_format, 'O'); in ShapeFromFilterFormat()
625 GetFilterDim(src_shape, src_filter_format, 'I') * in ShapeFromFilterFormat()
631 {{GetFilterDim(src_shape, src_filter_format, '0'), in ShapeFromFilterFormat()
632 GetFilterDim(src_shape, src_filter_format, '1'), in ShapeFromFilterFormat()
633 GetFilterDim(src_shape, src_filter_format, '2')}}, in ShapeFromFilterFormat()
[all …]
|