Searched refs:feature_index_ (Results 1 – 2 of 2) sorted by relevance
82 int64_t feature_index() const { return feature_index_; } in feature_index()119 int64_t feature_index_ = -1; variable
116 feature_index_(feature_index) { in HloBatchNormInstruction()133 proto.set_feature_index(feature_index_); in ToProto()