Searched defs:data_dims (Results 1 – 3 of 3) sorted by relevance
1638 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local1658 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local2332 std::vector<int64_t> data_dims = in TEST_F() local3198 std::vector<int64_t> data_dims = RandomDims(4, 4, 1, 8); in TEST_F() local3368 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local3454 std::vector<int64_t> data_dims = RandomDims(0, kDefaultMaxRank, 1); in TEST_F() local3468 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local3655 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local3956 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local4476 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local[all …]
329 ConvDimIndices GetDimIndices(const DataLayout& layout, const int data_dims) { in GetDimIndices()365 ConvDimIndices GetDimIndices(const FilterLayout& layout, const int data_dims) { in GetDimIndices()
151 int64_t data_dims = tile_assignment.num_dimensions() - subgroup_types.size(); in Subgroup() local