Home
last modified time | relevance | path

Searched refs:scale_idx (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc2215 int scale_idx = 1; in TrySimplify() local
2218 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()