Searched refs:channels_dim (Results 1 – 2 of 2) sorted by relevance
33 DimensionHandle channels_dim; in DecodeWavShapeFn() local37 channels_dim = c->UnknownDim(); in DecodeWavShapeFn()43 channels_dim = c->MakeDim(desired_channels); in DecodeWavShapeFn()57 c->set_output(0, c->MakeShape({samples_dim, channels_dim})); in DecodeWavShapeFn()
72 DimensionHandle channels_dim; in DecodeImageShapeFn() local76 channels_dim = c->UnknownDim(); in DecodeImageShapeFn()82 channels_dim = c->MakeDim(channels); in DecodeImageShapeFn()86 InferenceContext::kUnknownDim, channels_dim})); in DecodeImageShapeFn()94 DimensionHandle channels_dim; in DecodeImageV2ShapeFn() local101 channels_dim = c->UnknownDim(); in DecodeImageV2ShapeFn()107 channels_dim = c->MakeDim(channels); in DecodeImageV2ShapeFn()119 InferenceContext::kUnknownDim, channels_dim})); in DecodeImageV2ShapeFn()486 DimensionHandle channels_dim = c->UnknownDim(); in __anon30b5031d0802() local497 channels_dim = c->MakeDim(channels); in __anon30b5031d0802()[all …]