Searched defs:shape_1 (Results 1 – 5 of 5) sorted by relevance
42 std::vector<int> shape_1; in IsBroadcastable() local
545 def relax_input_shape(shape_1, shape_2): argument
121 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local150 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
798 const Shape& shape_1, in HasCompatibleElementTypes()811 const Shape& shape_1 = instruction->operand(1)->shape(); in HandleRng() local
2236 const int64_t shape_1[] = {1, 3}; in TEST_F() local