Searched defs:output_features (Results 1 – 3 of 3) sorted by relevance
52 std::vector<float> output_features; in GetCachedClickContextFeatures() local60 std::vector<float> output_features; in GetCachedBoundsSensitiveFeatures() local
230 auto output_features = in TryResolvePaddedShapesForTensorCore() local
511 int64_t output_features = rhs->shape().dimensions(output_feature_dimension); in MatchBackwardInput() local