HomeSort by relevance Sort by last modified time
    Searched defs:flat_input (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
relu_op.cc 198 auto flat_input = input.flat<qint8>(); local
199 OP_REQUIRES(context, (flat_input.size() % 4) == 0,
202 flat_input.size()));
204 func(context->eigen_device<Device>(), flat_input, output->flat<qint8>());
  /external/tensorflow/tensorflow/compiler/tf2xla/
fused_batchnorm_reserve_space_test.cc 113 auto flat_input = input_data.flat<float>(); local
114 for (int i = 0; i < flat_input.size(); i++) {
115 flat_input.data()[i] = (i - 5) / 1000.0f;

Completed in 221 milliseconds