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.cc287 dst_format == FORMAT_OIHW in operator ()()
304 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OIHW)); in operator ()()
Dconv_ops_3d.cc432 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in launch()
436 dst_format == FORMAT_OIHW in launch()
Dconv_grad_ops_3d.cc1334 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1336 dst_format == FORMAT_OIHW in Compute()
1830 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1832 dst_format == FORMAT_OIHW in Compute()
Dconv_2d_gpu.h496 if (dst_filter_format == FORMAT_OIHW) {
525 if (src_filter_format == FORMAT_OIHW) {
Dconv_2d.h228 if (dst_filter_format == FORMAT_OIHW) {
Dconv_ops.cc913 dst_format == FORMAT_OIHW in operator ()()
930 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OIHW)); in operator ()()
Dconv_ops_fused_impl.h612 context->eigen_device<GPUDevice>(), FORMAT_OIHW,
Dconv_grad_filter_ops.cc1111 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in operator ()()