Home
last modified time | relevance | path

Searched refs:ofm_dim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc320 int64_t ofm_dim = GetTensorSpatialDimIndex( in getTransposeConv2dPaddingValues() local
326 int64_t ofm_size = output_type.getDimSize(ofm_dim); in getTransposeConv2dPaddingValues()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md1042 int64_t ofm_dim = GetTensorSpatialDimIndex(4, data_format, i);
1046 int64_t ofm_size = output_dims[ofm_dim]