Searched refs:r1s2f32_ (Results 1 – 2 of 2) sorted by relevance
56 auto x = Parameter(&builder, 0, r1s2f32_, "x"); in CreateR1S2F32AdditionComputation()57 auto y = Parameter(&builder, 1, r1s2f32_, "y"); in CreateR1S2F32AdditionComputation()74 Shape r1s2f32_ = ShapeUtil::MakeShape(F32, {2}); member in xla::__anon17d4f22b0111::CallOpTest
59 return CreateCeilComputation(r1s2f32_); in CreateR1CeilComputation()76 return CreateFloorComputation(r1s2f32_); in CreateR1FloorComputation()164 Shape r1s2f32_ = ShapeUtil::MakeShape(F32, {2}); member in xla::__anonf9fe2de30111::ConditionalOpTest