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

  /external/tensorflow/tensorflow/core/util/
mkl_util_test.cc 44 memory::dims b_dims = {N, C, H, W}; local
48 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, memory::format::nhwc);
  /external/tensorflow/tensorflow/python/kernel_tests/
tensordot_op_test.py 145 b_dims = _random_subset(num_dims_, rank_b_)
148 b_shape[b_dims[i]] = shared_shape[i]
155 return a, b, a_dims, b_dims
  /external/tensorflow/tensorflow/compiler/tests/
randomized_tests.cc     [all...]

Completed in 228 milliseconds