HomeSort by relevance Sort by last modified time
    Searched refs:in_flat (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
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...]
unary_ops_composition.cc 113 InputBuffer in_flat = in.flat<T>(); variable
117 auto compute_fn = [this, &in_flat, &out_flat, &num_fns](int64 begin,
120 const InputBuffer in_slice(in_flat.data() + begin, len);
pooling_ops_common.h 520 auto in_flat = input.flat<T>(); local
523 auto shard = [&params, &in_flat, &out_flat](int64 start, int64 limit) {
532 in_flat.data() + start * input_image_size, params.depth,

Completed in 109 milliseconds