Searched refs:in_flat (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fake_quant_ops_test.cc | 669 auto in_flat = GetInput(0).flat<float>(); in TEST_F() local
672 {in_flat(0), in_flat(1), in_flat(2), in_flat(3), in_flat(4), in_flat(5)}); in TEST_F()
714 auto in_flat = GetInput(0).flat<float>(); in TEST_F() local
715 FillValues<float>(&expected_bprop_wrt_input, {0.0f, in_flat(1), in_flat(2), in TEST_F()
716 in_flat(3), in_flat(4), 0.0f}); in TEST_F()
721 expected_bprop_wrt_min.flat<float>()(0) = in_flat(0); in TEST_F()
726 expected_bprop_wrt_max.flat<float>()(0) = in_flat(5); in TEST_F()
758 auto in_flat = GetInput(0).flat<float>(); in TEST_F() local
759 FillValues<float>(&expected_bprop_wrt_input, {0.0f, in_flat(1), in_flat(2), in TEST_F()
760 in_flat(3), in_flat(4), 0.0f}); in TEST_F()
[all …]
|
| D | unary_ops_composition.cc | 113 InputBuffer in_flat = in.flat<T>(); in Compute() local
117 auto compute_fn = [this, &in_flat, &out_flat, &num_fns](int64 begin, in Compute()
120 const InputBuffer in_slice(in_flat.data() + begin, len); in Compute()
|
| D | pooling_ops_common.h | 520 auto in_flat = input.flat<T>();
523 auto shard = [¶ms, &in_flat, &out_flat](int64 start, int64 limit) {
532 in_flat.data() + start * input_image_size, params.depth,
|