Searched refs:b_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 43 memory::dims b_dims = {N, C, H, W}; in TEST() local
47 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, MKL_TENSOR_FORMAT_NHWC); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | triangular_solve_test.cc | 471 std::vector<int64> b_dims = spec.dims; in XLA_TEST_P() local
473 std::swap(b_dims.back(), b_dims.at(b_dims.size() - 2)); in XLA_TEST_P()
475 Array<float> bvals(b_dims); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 513 memory::dims b_dims; member
519 MklMatMulParams(memory::dims a_dims, memory::dims b_dims, memory::dims c_dims, in MklMatMulParams()
523 b_dims(b_dims), in MklMatMulParams()
603 new memory::desc({params.b_dims}, MklDnnType<T>(), params.b_strides)); in Setup()
673 key_creator.AddAsKey(params.b_dims); in CreateKey()
703 dims b_dims = dims{k, n};
714 MklMatMulParams params(a_dims, b_dims, c_dims, a_strides, b_strides,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensordot_op_test.py | 155 b_dims = _random_subset(num_dims_, rank_b_)
158 b_shape[b_dims[i]] = shared_shape[i]
165 return a, b, a_dims, b_dims
|
| D | eig_op_test.py | 232 b_dims = len(e.shape) - 1
234 e = array_ops.gather(e, idx, batch_dims=b_dims)
235 v = array_ops.gather(v, idx, batch_dims=b_dims)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2246 std::vector<int64> b_dims = {y, z}; in TEST_F() local
2255 std::swap(b_dims[0], b_dims[1]); in TEST_F()
2261 .RandomInput(type, b_dims) in TEST_F()
3060 std::vector<int64> b_dims = {y, z}; in TEST_F() local
3069 std::swap(b_dims[0], b_dims[1]); in TEST_F()
3074 .RandomInput(DT_FLOAT, b_dims) in TEST_F()
|