Searched refs:FilterTensorFormat (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 75 enum FilterTensorFormat { enum
101 FilterTensorFormat* format);
107 string ToString(FilterTensorFormat format);
126 inline int GetFilterTensorSpatialDims(int num_dims, FilterTensorFormat format) { in GetFilterTensorSpatialDims()
153 FilterTensorFormat format) { in GetFilterTensorDimsFromSpatialDims()
237 FilterTensorFormat format, int dim) { in GetFilterTensorSpatialDimIndex()
254 int num_dims, FilterTensorFormat format) { in GetFilterTensorInnerInputChannelsDimIndex()
263 FilterTensorFormat format) { in GetFilterTensorInputChannelsDimIndex()
278 FilterTensorFormat format) { in GetFilterTensorOutputChannelsDimIndex()
370 inline int GetFilterDimIndex(FilterTensorFormat filter_tensor_format, in GetFilterDimIndex()
[all …]
|
| D | tensor_format_test.cc | 34 std::pair<FilterTensorFormat, const char*> test_filter_formats[] = {
128 const FilterTensorFormat format) { in GetFilterDimMap()
162 FilterTensorFormat format; in TEST()
189 FilterTensorFormat format = test_filter_format.first; in RunDimensionIndexesTest()
|
| D | tensor_format.cc | 60 string ToString(FilterTensorFormat format) { in ToString()
103 FilterTensorFormat* format) { in FilterFormatFromString()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.h | 45 FilterTensorFormat filter_format, Tensor* output);
59 FilterTensorFormat filter_format, Tensor* output);
|
| D | fused_conv2d_bias_activation_op.cc | 252 FilterTensorFormat filter_format_;
447 FilterTensorFormat filter_format, Tensor* output_param) { in launch()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| D | fused_conv2d_bias_activation_op.cc | 60 FilterTensorFormat filter_format; in __anon0cecbc340102()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d.h | 153 void operator()(const Device& d, FilterTensorFormat dst_filter_format,
|
| D | conv_ops_3d.cc | 519 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
|
| D | conv_2d_gpu.h | 425 void operator()(const Device& d, FilterTensorFormat dst_filter_format,
|
| D | conv_ops_fused_impl.h | 970 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
|
| D | conv_ops.cc | 940 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
|
| D | conv_grad_filter_ops.cc | 927 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
|
| D | conv_grad_input_ops.cc | 1070 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
|
| D | conv_grad_ops_3d.cc | 1061 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 425 FilterTensorFormat filter_format; in Conv2DShapeImpl()
|