Home
last modified time | relevance | path

Searched defs:b_dims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc42 memory::dims b_dims = {N, C, H, W}; in TEST() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtriangular_solve_test.cc464 std::vector<int64_t> b_dims = spec.dims; in XLA_TEST_P() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_ops_common.h586 memory::dims b_dims; member
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3240 std::vector<int64_t> b_dims = {y, z}; in TEST_F() local
4305 std::vector<int64_t> b_dims = {y, z}; in TEST_F() local