Searched refs:FORMAT_OIHW (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.cc | 64 case FORMAT_OIHW: in ToString()
109 *format = FORMAT_OIHW; in FilterFormatFromString()
|
| D | tensor_format.h | 81 FORMAT_OIHW = 1, enumerator
243 case FORMAT_OIHW: in GetFilterTensorSpatialDimIndex()
267 case FORMAT_OIHW: in GetFilterTensorInputChannelsDimIndex()
282 case FORMAT_OIHW: in GetFilterTensorOutputChannelsDimIndex()
386 } else if (filter_tensor_format == FORMAT_OIHW || in GetFilterDimIndex()
|
| D | tensor_format_test.cc | 36 EnumStringPair(FORMAT_OIHW),
131 (format == FORMAT_OIHW || in GetFilterDimMap()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 629 FORMAT_OIHW, filter_param.shape(), FORMAT_HWIO), in launch()
632 ctx->eigen_device<GPUDevice>(), FORMAT_OIHW, in launch()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d.h | 165 DCHECK(dst_filter_format == FORMAT_OIHW)
|
| D | conv_ops_3d.cc | 391 ctx->eigen_device<GPUDevice>(), FORMAT_OIHW, in launch()
|
| D | conv_2d_gpu.h | 437 CHECK(dst_filter_format == FORMAT_OIHW)
|
| D | conv_ops_fused_impl.h | 718 context->eigen_device<GPUDevice>(), FORMAT_OIHW,
|
| D | conv_ops.cc | 796 ctx->eigen_device<GPUDevice>(), FORMAT_OIHW, in operator ()()
|
| D | conv_grad_input_ops.cc | 880 ctx->eigen_device<GPUDevice>(), FORMAT_OIHW, in operator ()()
|
| D | conv_grad_ops_3d.cc | 1295 context->eigen_device<GPUDevice>(), FORMAT_OIHW, in Compute()
|