Searched defs:channel_dim (Results 1 – 6 of 6) sorted by relevance
31 int size_input_idx, DimensionHandle channel_dim) { in SetOutputToSizedImage()
103 int channel_dim = input_shape.dims() - 1; in Compile() local141 int channel_dim = input_shape.dims() - 1; in Compile() local181 int channel_dim = input_shape.dims() - 1; in Compile() local231 const int channel_dim = input_shape.dims() - 1; in Compile() local285 const int channel_dim = input_shape.dims() - 1; in Compile() local
55 int32 channel_dim = value_tensor.dims() - 1; in GetBiasValueDims() local112 size_t channel_dim; in Compute() local
2395 DimensionHandle channel_dim = in __anon3e672dd83e02() local2462 DimensionHandle channel_dim = (data_format == "NHWC") in __anon3e672dd83f02() local
804 DimensionHandle channel_dim = c->Dim(x, channel_dim_index); in FusedBatchNormShape() local840 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape() local