Lines Matching defs:feature_index
56 int feature_index =
66 input, ctx->Input(1), ctx->Input(2), epsilon_, feature_index);
84 epsilon_, feature_index);
130 int feature_index =
152 epsilon_, feature_index);
160 std::iota(reduction_dims.begin(), reduction_dims.begin() + feature_index,
162 std::iota(reduction_dims.begin() + feature_index, reduction_dims.end(),
163 feature_index + 1);
179 b->Mul(grad_backprop, b->Sub(activations, mean, {feature_index})),
184 b->Mul(grad_backprop, b->Mul(scratch1, scale), {feature_index});
