Searched defs:offset_shape (Results 1 – 5 of 5) sorted by relevance
65 … const std::vector<int> &offset_shape, const std::vector<int> &output_shape, in Prepare()101 std::vector<int> offset_shape{3}; in TEST_F() local123 std::vector<int> offset_shape{3}; in TEST_F() local144 std::vector<int> offset_shape{3}; in TEST_F() local
223 std::vector<size_t> offset_shape = {1}; in HashSwapOut() local264 std::vector<size_t> offset_shape = {1}; in HashSwapIn() local
107 const std::vector<int> &offset_shape) { in ScaleInputShapeValid()
134 GpuTensorInfo offset_shape = GpuTensorInfo(offset_tensor); in InitWeights() local
95 def check_shape(self, input_x_shape, max_num_shape, offset_shape): argument