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

  /external/tensorflow/tensorflow/core/kernels/
qr_op_impl.h 174 Tensor input_transposed; variable
182 transposed_shape, &input_transposed),
196 context, DoMatrixTranspose(device, input, &input_transposed), done);
198 // Compute QR decomposition in-place in input_transposed.
202 input_transposed.flat_inner_dims<Scalar, 3>();
215 // stored in input_transposed. Crop, transpose (to get back to row-major)
219 context, DoMatrixTranspose(device, input_transposed, r), done);
242 // Generate Q from the decomposition in input_transposed.
281 context, DoMatrixTranspose(device, input_transposed, q), done);
  /external/tensorflow/tensorflow/contrib/lite/kernels/
transpose_test.cc 30 std::vector<float>* input_transposed) {
34 input_transposed->resize(count);
67 input_transposed->data(), output_dims,

Completed in 90 milliseconds