Searched refs:parameter_shapes (Results 1 – 2 of 2) sorted by relevance
61 std::vector<const ShapeProto*> parameter_shapes; in EntryComputationParameterShapes() local64 parameter_shapes.push_back(&shape); in EntryComputationParameterShapes()66 return parameter_shapes; in EntryComputationParameterShapes()
355 std::vector<Shape> parameter_shapes; in XLA_TEST_F() local362 parameter_shapes.push_back(literal.shape()); in XLA_TEST_F()373 parameter_shapes.push_back(bool_literal.shape()); in XLA_TEST_F()378 Shape while_shape = ShapeUtil::MakeTupleShape(parameter_shapes); in XLA_TEST_F()