Searched refs:FORMAT_HWIO (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.cc | 62 case FORMAT_HWIO: in ToString()
107 *format = FORMAT_HWIO; in FilterFormatFromString()
|
| D | tensor_format.h | 80 FORMAT_HWIO = 0, enumerator
252 case FORMAT_HWIO: in GetFilterTensorSpatialDimIndex()
276 case FORMAT_HWIO: in GetFilterTensorInputChannelsDimIndex()
291 case FORMAT_HWIO: in GetFilterTensorOutputChannelsDimIndex()
384 if (filter_tensor_format == FORMAT_HWIO) { in GetFilterDimIndex()
|
| D | tensor_format_test.cc | 35 EnumStringPair(FORMAT_HWIO),
130 (format == FORMAT_HWIO) ? DimMaps::kFdmHWIO[num_spatial_dims] : in GetFilterDimMap()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 283 VLOG(4) << "Transform filter tensor from " << ToString(FORMAT_HWIO) in operator ()()
|
| D | conv_ops_3d.cc | 433 VLOG(4) << "Transform filter tensor from " << ToString(FORMAT_HWIO) in launch()
|
| D | conv_ops.cc | 909 VLOG(4) << "Transform filter tensor from " << ToString(FORMAT_HWIO) in operator ()()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1533 num_dims, tensorflow::FORMAT_HWIO); in Verify()
1643 num_dims, tensorflow::FORMAT_HWIO)); in inferConvReturnTypes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1510 …om_pad_type(%input_sizes, %filter, output_shape, padding, data_format, FORMAT_HWIO, strides, dilat…
1542 FORMAT_HWIO, strides, dilations)
1598 FORMAT_HWIO, strides, dilations)
|