Home
last modified time | relevance | path

Searched refs:FilterTensorFormat (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h77 enum FilterTensorFormat { enum
106 FilterTensorFormat* format);
112 std::string ToString(FilterTensorFormat format);
134 inline int GetFilterTensorSpatialDims(int num_dims, FilterTensorFormat format) { in GetFilterTensorSpatialDims()
164 FilterTensorFormat format) { in GetFilterTensorDimsFromSpatialDims()
248 FilterTensorFormat format, int dim) { in GetFilterTensorSpatialDimIndex()
265 int num_dims, FilterTensorFormat format) { in GetFilterTensorInnerInputChannelsDimIndex()
274 FilterTensorFormat format) { in GetFilterTensorInputChannelsDimIndex()
289 FilterTensorFormat format) { in GetFilterTensorOutputChannelsDimIndex()
381 inline int GetFilterDimIndex(FilterTensorFormat filter_tensor_format, in GetFilterDimIndex()
[all …]
Dtensor_format_test.cc34 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()
Dtensor_format.cc60 string ToString(FilterTensorFormat format) { in ToString()
105 FilterTensorFormat* format) { in FilterFormatFromString()
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h208 void operator()(const Device& d, FilterTensorFormat dst_filter_format,
397 void operator()(const Device& d, FilterTensorFormat src_filter_format,
Dconv_grad_input_ops.cc282 const auto transform_filter = [&](FilterTensorFormat dst_format) -> Status { in operator ()()
557 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
Dconv_ops_3d.cc652 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
657 const GPUDevice& d, FilterTensorFormat src_filter_format, \
Dconv_2d_gpu.h484 void operator()(const Device& d, FilterTensorFormat dst_filter_format,
520 void operator()(const Device& d, FilterTensorFormat src_filter_format,
Dconv_grad_filter_ops.cc1110 FilterTensorFormat src_filter_format = in operator ()()
1125 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
Dconv_ops.cc908 const auto transform_filter = [&](FilterTensorFormat dst_format) -> Status { in operator ()()
1163 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
Dconv_ops_fused_impl.h824 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
Dconv_grad_ops_3d.cc1072 const GPUDevice& d, FilterTensorFormat dst_filter_format, \
1077 const GPUDevice& d, FilterTensorFormat src_filter_format, \
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc592 FilterTensorFormat filter_format; in Conv2DShapeImpl()