Searched refs:input_depth_dim (Results 1 – 1 of 1) sorted by relevance
637 DimensionHandle input_depth_dim; in Conv2DShapeImpl() local641 absl::MakeSpan(input_spatial_dims), &input_depth_dim, c)); in Conv2DShapeImpl()664 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv2DShapeImpl()665 int64 input_depth_value = c->Value(input_depth_dim), in Conv2DShapeImpl()793 DimensionHandle input_depth_dim = c->Dim(input_shape, 4); in Conv3DShape() local803 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv3DShape()804 int64 input_depth_value = c->Value(input_depth_dim), in Conv3DShape()