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

  /external/tensorflow/tensorflow/compiler/tests/
test_utils.py 42 transpose_dims = [dim_map[d] for d in data_format_dst]
43 return np.transpose(x, transpose_dims)
  /external/tensorflow/tensorflow/compiler/xla/client/lib/
matrix.cc 243 std::vector<int64> transpose_dims(output_rank);
245 transpose_dims[output_dims[i]] = i;
251 return Transpose(DotGeneral(x, y, dnums, &precision_proto), transpose_dims);
  /external/tensorflow/tensorflow/python/ops/
array_grad.py 463 transpose_dims = array_ops.concat(
465 values_transpose = array_ops.transpose(values, transpose_dims)
    [all...]

Completed in 377 milliseconds