Searched defs:channel_dim (Results 1 – 6 of 6) sorted by relevance
105 int channel_dim = input_shape.dims() - 1; in Compile() local143 int channel_dim = input_shape.dims() - 1; in Compile() local183 int channel_dim = input_shape.dims() - 1; in Compile() local233 const int channel_dim = input_shape.dims() - 1; in Compile() local287 const int channel_dim = input_shape.dims() - 1; in Compile() local
54 int32 channel_dim = value_tensor.dims() - 1; in GetBiasValueDims() local111 size_t channel_dim; in Compute() local
50 int channel_dim; member
1153 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormShape() local1193 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormExShape() local1223 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape() local
31 int size_input_idx, DimensionHandle channel_dim) { in SetOutputToSizedImage()
820 const uint64_t channel_dim = rank - 1; in matchAndRewrite() local