Lines Matching refs:feature_index
383 HloInstruction* offset, float epsilon, int64 feature_index);
389 float epsilon, int64 feature_index);
395 HloInstruction* grad_output, float epsilon, int64 feature_index);
827 // Returns feature_index field associated with the instruction. The index
832 int64 feature_index() const { return feature_index_; }