Searched defs:pos_shaped (Results 1 – 1 of 1) sorted by relevance
152 auto pos_shaped = pos_tensor.shaped<T, 1>(bcast.y_reshape()); in Compute() local205 auto pos_shaped = pos_tensor.shaped<T, 2>(bcast.y_reshape()); in Compute() local