Searched refs:a_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 31 memory::dims a_dims = {N, C, H, W}; in TEST() local
35 a_mkldnn_shape.SetTfLayout(a_dims.size(), a_dims, MKL_TENSOR_FORMAT_NCHW); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | triangular_solve_test.cc | 461 std::vector<int64> a_dims = spec.dims; in XLA_TEST_P() local
462 a_dims.back() = a_dims.at(a_dims.size() - 2); in XLA_TEST_P()
463 Array<float> avals(a_dims); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 512 memory::dims a_dims; member
519 MklMatMulParams(memory::dims a_dims, memory::dims b_dims, memory::dims c_dims, in MklMatMulParams()
522 : a_dims(a_dims), in MklMatMulParams()
600 new memory::desc({params.a_dims}, MklDnnType<T>(), params.a_strides)); in Setup()
672 key_creator.AddAsKey(params.a_dims); in CreateKey()
702 dims a_dims = dims{m, k};
714 MklMatMulParams params(a_dims, b_dims, c_dims, a_strides, b_strides,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensordot_op_test.py | 154 a_dims = _random_subset(num_dims_, rank_a_)
157 a_shape[a_dims[i]] = shared_shape[i]
165 return a, b, a_dims, b_dims
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2245 std::vector<int64> a_dims = {x, y}; in TEST_F() local
2252 std::swap(a_dims[0], a_dims[1]); in TEST_F()
2260 .RandomInput(type, a_dims) in TEST_F()
3059 std::vector<int64> a_dims = {x, y}; in TEST_F() local
3066 std::swap(a_dims[0], a_dims[1]); in TEST_F()
3073 .RandomInput(DT_FLOAT, a_dims) in TEST_F()
|