Lines Matching defs:feature_index
849 optional<int64> feature_index;
850 attrs["feature_index"] = {/*required=*/true, AttrTy::kInt64,
851 &feature_index};
859 /*offset=*/operands[2], *epsilon, *feature_index));
865 optional<int64> feature_index;
866 attrs["feature_index"] = {/*required=*/true, AttrTy::kInt64,
867 &feature_index};
876 /*variance=*/operands[4], *epsilon, *feature_index));
882 optional<int64> feature_index;
883 attrs["feature_index"] = {/*required=*/true, AttrTy::kInt64,
884 &feature_index};
892 /*grad_output=*/operands[4], *epsilon, *feature_index));
