Searched refs:GetFilterDim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 435 T GetFilterDim(gtl::ArraySlice<T> dimension_attribute, in GetFilterDim() function
463 inline int64 GetFilterDim(const TensorShape& tensor_shape, in GetFilterDim() function
466 return GetFilterDim(gtl::ArraySlice<int64>(tensor_shape.dim_sizes()), in GetFilterDim()
479 inline int64 GetFilterDim(const Tensor& tensor, in GetFilterDim() function
482 return GetFilterDim(tensor.shape(), filter_tensor_format, dimension); in GetFilterDim()
610 const int64 output_channels = GetFilterDim(src_shape, src_filter_format, 'O'); in ShapeFromFilterFormat()
612 GetFilterDim(src_shape, src_filter_format, 'I') * in ShapeFromFilterFormat()
618 {{GetFilterDim(src_shape, src_filter_format, '0'), in ShapeFromFilterFormat()
619 GetFilterDim(src_shape, src_filter_format, '1'), in ShapeFromFilterFormat()
620 GetFilterDim(src_shape, src_filter_format, '2')}}, in ShapeFromFilterFormat()
[all …]
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 203 const int32 filter_rows = GetFilterDim(filter, filter_format_, 'H'); in Compute()
204 const int32 filter_cols = GetFilterDim(filter, filter_format_, 'W'); in Compute()
205 const int32 output_depth = GetFilterDim(filter, filter_format_, 'O'); in Compute()
478 const int output_depth = GetFilterDim(filter_param, filter_format, 'O'); in launch()
479 const int filter_rows = GetFilterDim(filter_param, filter_format, 'H'); in launch()
480 const int filter_cols = GetFilterDim(filter_param, filter_format, 'W'); in launch()
|