Searched defs:in_cpu (Results 1 – 1 of 1) sorted by relevance
129 Tensor in_cpu(data_type_, shape); in MakeReductionTestCase() local178 Tensor in_cpu(data_type_, in_shape); in MakeGatherTestCase() local229 Tensor in_cpu(data_type_, shape); in MakeBroadcastTestCase() local