Searched refs:feature_index_ (Results 1 – 2 of 2) sorted by relevance
32 int64 feature_index() const { return feature_index_; } in feature_index()58 int64 feature_index_ = -1; variable
81 feature_index_(feature_index) { in HloBatchNormInstruction()98 proto.set_feature_index(feature_index_); in ToProto()