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

  /external/tensorflow/tensorflow/core/kernels/
quantization_utils_test.cc 47 Tensor i_tensor = local
53 auto input_array = i_tensor.flat<qint32>();
59 *eigen_device, i_tensor, input_min, input_max, output_min, output_max,
89 const Tensor i_tensor = local
94 const auto input_array = i_tensor.flat<quint8>();
244 Tensor i_tensor = tensorflow::test::AsTensor( local
246 const auto i_array = i_tensor.flat<InputType>();
258 eigen_device, i_tensor, input_min, input_max, output_min, output_max,
300 Tensor i_tensor = local
302 const auto i_array = i_tensor.flat<InputType>()
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.cc 303 Tensor* i_tensor = nullptr; variable
306 TensorShape({batch_size, cell_size}), &i_tensor));
359 xh_tensor.matrix<T>(), i_tensor->matrix<T>(), cs_tensor->matrix<T>(),
444 const Tensor* i_tensor = nullptr; variable
445 OP_REQUIRES_OK(ctx, ctx->input("i", &i_tensor));
505 OP_REQUIRES(ctx, i_tensor->dim_size(0) == batch_size,
507 "i.dim_size(0) != batch_size: ", i_tensor->dim_size(0),
509 OP_REQUIRES(ctx, i_tensor->dim_size(1) == cell_size,
511 "i.dim_size(1) != cell_size: ", i_tensor->dim_size(1),
641 wco_tensor->vec<T>(), b_tensor->vec<T>(), i_tensor->matrix<T>()
962 Tensor i_tensor = slicer.OutputSlice(i_out, t, "i_out"); variable
1218 const Tensor& i_tensor = slicer.InputSlice(*i_out, t, "i_out"); variable
    [all...]

Completed in 66 milliseconds