Searched defs:a_dims (Results 1 – 5 of 5) sorted by relevance
29 memory::dims a_dims = {N, C, H, W}; in TEST() local
534 absl::InlinedVector<int64_t, 4>& a_dims, in RemoveTrivialDimensions()590 absl::InlinedVector<int64_t, 4>& a_dims, in CoalesceDimensions()
454 std::vector<int64_t> a_dims = spec.dims; in XLA_TEST_P() local
585 memory::dims a_dims; member
3239 std::vector<int64_t> a_dims = {x, y}; in TEST_F() local4304 std::vector<int64_t> a_dims = {x, y}; in TEST_F() local