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

  /external/tensorflow/tensorflow/core/kernels/
sparse_matmul_op.cc 798 bool transpose_output, MatrixMap* output);
808 bool transpose_output, MatrixMap* output);
1026 bool transpose_output = false; variable
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/lib/
quantize.h 77 // If transpose_output is true, will return a tensor of shape
78 // [dn * unpack_size, dn-1, ..., d1, d0]. transpose_output is faster when
80 // transpose_output feature.
84 bool transpose_output = false) {
170 // Return the transpose result if transpose_output is true.
171 if (transpose_output) {
quantize_test.cc 81 bool transpose_output = false) {
90 if (transpose_output) {
117 if (transpose_output) {
238 xla::Dequantize<uint8>(x, range, "MIN_COMBINED", /*transpose_output=*/true);
293 xla::Dequantize<uint8>(x, range, "MIN_COMBINED", /*transpose_output=*/true);
329 xla::Dequantize<uint8>(x, range, "MIN_COMBINED", /*transpose_output=*/true);
332 input, range, /*transpose_output=*/true);
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 625 const string transpose_output = local
628 AvailableArrayName(model, transpose_output + "/perm");
632 transpose_op->set_name(transpose_output);
    [all...]

Completed in 213 milliseconds