Lines Matching refs:num_slots
135 int64 num_slots = accumulator_resource.values().size();
138 TensorShape({num_slots}),
145 TensorShape({num_slots, 2}),
152 "output_gradients", TensorShape({num_slots}), &gradients_t));
158 TensorShape({num_slots}), &hessians_t));
176 int64 num_slots = accumulator_resource.values().size();
179 TensorShape({num_slots}),
185 TensorShape({num_slots, 2}),
191 gradient_shape.InsertDim(0, num_slots);
200 hessian_shape.InsertDim(0, num_slots);
