Searched refs:feature_index_ (Results 1 – 4 of 4) sorted by relevance
91 feature_index_(feature_index), in CudnnBatchNormForwardInferenceThunk()107 MakeDescriptors(hlo_instruction()->shape(), feature_index_); in ExecuteOnStream()148 feature_index_(feature_index), in CudnnBatchNormForwardTrainingThunk()172 hlo_instruction()->shape().tuple_shapes(0), feature_index_); in ExecuteOnStream()230 feature_index_(feature_index), in CudnnBatchNormBackwardThunk()257 hlo_instruction()->shape().tuple_shapes(0), feature_index_); in ExecuteOnStream()
74 int64 feature_index_; variable103 int64 feature_index_; variable139 int64 feature_index_; variable
30 int64 feature_index() const { return feature_index_; } in feature_index()56 int64 feature_index_ = -1; variable
78 feature_index_(feature_index) { in HloBatchNormInstruction()95 proto.set_feature_index(feature_index_); in ToProto()