Searched refs:scale_idx (Results 1 – 1 of 1) sorted by relevance
2215 int scale_idx = 1; in TrySimplify() local2218 TF_RETURN_IF_ERROR(GetInputNode(mul->input(scale_idx), &scale)); in TrySimplify()2222 std::swap(scale_idx, input_idx); in TrySimplify()2254 scaled_weights->add_input(mul->input(scale_idx)); in TrySimplify()