fake_quant_ops_test.cc | 669 auto in_flat = GetInput(0).flat<float>(); local 672 {in_flat(0), in_flat(1), in_flat(2), in_flat(3), in_flat(4), in_flat(5)}); 714 auto in_flat = GetInput(0).flat<float>(); local 715 FillValues<float>(&expected_bprop_wrt_input, {0.0f, in_flat(1), in_flat(2) 758 auto in_flat = GetInput(0).flat<float>(); local 803 auto in_flat = GetInput(0).flat<float>(); local 848 auto in_flat = GetInput(0).flat<float>(); local [all...] |