Home
last modified time | relevance | path

Searched refs:FORMAT_OIHW (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_format.cc64 case FORMAT_OIHW: in ToString()
111 *format = FORMAT_OIHW; in FilterFormatFromString()
Dtensor_format.h83 FORMAT_OIHW = 1, enumerator
254 case FORMAT_OIHW: in GetFilterTensorSpatialDimIndex()
278 case FORMAT_OIHW: in GetFilterTensorInputChannelsDimIndex()
293 case FORMAT_OIHW: in GetFilterTensorOutputChannelsDimIndex()
397 } else if (filter_tensor_format == FORMAT_OIHW || in GetFilterDimIndex()
Dtensor_format_test.cc36 EnumStringPair(FORMAT_OIHW),
131 (format == FORMAT_OIHW || in GetFilterDimMap()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc283 dst_format == FORMAT_OIHW in operator ()()
300 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OIHW)); in operator ()()
Dconv_ops_3d.cc440 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in launch()
444 dst_format == FORMAT_OIHW in launch()
Dconv_ops_fused_impl.h531 dst_format == FORMAT_OIHW
550 OP_REQUIRES_OK(context, transform_filter(FORMAT_OIHW));
Dconv_grad_ops_3d.cc1436 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1438 dst_format == FORMAT_OIHW in Compute()
1834 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1836 dst_format == FORMAT_OIHW in Compute()
Dconv_2d_gpu.h496 if (dst_filter_format == FORMAT_OIHW) {
525 if (src_filter_format == FORMAT_OIHW) {
Dconv_ops.cc1064 dst_format == FORMAT_OIHW in operator ()()
1081 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OIHW)); in operator ()()
Dconv_2d.h320 if (dst_filter_format == FORMAT_OIHW) {
Dconv_grad_filter_ops.cc1007 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in operator ()()