Home
last modified time | relevance | path

Searched defs:a_dims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc29 memory::dims a_dims = {N, C, H, W}; in TEST() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtranspose.cc534 absl::InlinedVector<int64_t, 4>& a_dims, in RemoveTrivialDimensions()
590 absl::InlinedVector<int64_t, 4>& a_dims, in CoalesceDimensions()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtriangular_solve_test.cc454 std::vector<int64_t> a_dims = spec.dims; in XLA_TEST_P() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_ops_common.h585 memory::dims a_dims; member
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3239 std::vector<int64_t> a_dims = {x, y}; in TEST_F() local
4304 std::vector<int64_t> a_dims = {x, y}; in TEST_F() local