Searched refs:FORMAT_OHWI (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.cc | 66 case FORMAT_OHWI: in ToString()
|
| D | tensor_format.h | 86 FORMAT_OHWI = 2, enumerator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu.h | 502 } else if (dst_filter_format == FORMAT_OHWI) {
539 } else if (src_filter_format == FORMAT_OHWI) {
|
| D | conv_2d.h | 237 } else if (dst_filter_format == FORMAT_OHWI) {
|
| D | conv_grad_input_ops.cc | 306 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OHWI)); in operator ()()
|
| D | conv_ops_3d.cc | 432 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in launch()
|
| D | conv_grad_ops_3d.cc | 1334 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1830 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
|
| D | conv_grad_filter_ops.cc | 1111 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in operator ()()
|
| D | conv_ops.cc | 932 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OHWI)); in operator ()()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1322 FORMAT_OHWI, strides, {1, 1, 1, 1})
2194 FORMAT_OHWI, strides, {1, 1, 1, 1})
|